Add txt files to setup.py for CBIS-DDSM dataset.

PiperOrigin-RevId: 253190966
liwohj · Jun 14, 2019 · 27f1e7f · 27f1e7f
1 parent aa5ab66
commit 27f1e7f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -79,6 +79,11 @@
 # Static files needed by datasets.
 DATASET_FILES = [
     'image/caltech101_labels.txt',
+    'image/cbis_ddsm_calc_distributions.txt',
+    'image/cbis_ddsm_calc_types.txt',
+    'image/cbis_ddsm_mass_margins.txt',
+    'image/cbis_ddsm_mass_shapes.txt',
+    'image/cbis_ddsm_patch_labels.txt',
     'image/dtd_key_attributes.txt',
     'image/imagenet2012_labels.txt',
     'image/imagenet2012_validation_labels.txt',