Add package_data in setup.py for loading config and model files.

usnistgov · Jan 10, 2023 · 5f813b0 · 5f813b0
1 parent ab0bf81
commit 5f813b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/alignn/__init__.py b/alignn/__init__.py
@@ -1,2 +1,2 @@
 """Version number."""
-__version__ = "2023.01.09"
+__version__ = "2023.01.10"
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setuptools.setup(
     name="alignn",
-    version="2023.01.09",
+    version="2023.01.10",
     author="Kamal Choudhary, Brian DeCost",
     author_email="[email protected]",
     description="alignn",
@@ -33,7 +33,7 @@
         "ase",
         # "dgl-cu101>=0.6.0",
     ],
-    # scripts=["alignn/alignn_train_folder.py"],
+    package_data={"alignn.ff": ["best_model.pt", "config.json"]},
     scripts=[
         "alignn/pretrained.py",
         "alignn/train_folder.py",