Fix pruned txt files not being installed during pip install

rwightman · rwightman · commit 375f3e532184 · 2020-05-03T19:06:14.000-07:00
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1,2 @@
+include timm/models/pruned/*.txt
+
diff --git a/setup.py b/setup.py
@@ -41,6 +41,7 @@
     # Note that this is a string of words separated by whitespace, not a list.
     keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet',
     packages=find_packages(exclude=['convert']),
+    include_package_data=True,
     install_requires=['torch >= 1.0', 'torchvision'],
     python_requires='>=3.6',
 )

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+include timm/models/pruned/*.txt`
	`2`	`+`
Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,7 @@`
`41`	`41`	`# Note that this is a string of words separated by whitespace, not a list.`
`42`	`42`	`keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet',`
`43`	`43`	`packages=find_packages(exclude=['convert']),`
	`44`	`+ include_package_data=True,`
`44`	`45`	`install_requires=['torch >= 1.0', 'torchvision'],`
`45`	`46`	`python_requires='>=3.6',`
`46`	`47`	`)`