add mvl_models in wheel

materialsvirtuallab · Apr 14, 2020 · bd1432a · bd1432a
1 parent b2cfb37
commit bd1432a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mvl_models/__init__.py b/mvl_models/__init__.py
diff --git a/setup.py b/setup.py
@@ -27,7 +27,8 @@
     },
     packages=find_packages(),
     package_data={
-        "megnet": ["*.json", "*.md"]
+        "megnet": ["*.json", "*.md"],
+        "mvl_models": ["*/*.json", "*/*.md", "*/*.hdf5"],
     },
     keywords=["materials", "science", "machine", "learning", "deep", "graph", "networks", "neural"],
     classifiers=[