From 5f813b08b086d55e3cd210c0848c52a1200b8381 Mon Sep 17 00:00:00 2001 From: JARVIS-Unifies Date: Tue, 10 Jan 2023 17:02:19 -0500 Subject: [PATCH] Add package_data in setup.py for loading config and model files. --- alignn/__init__.py | 2 +- setup.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/alignn/__init__.py b/alignn/__init__.py index ed67e2c1..e8c5f63f 100644 --- 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 index 65a467b7..b2f0443c 100644 --- 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="kamal.choudhary@nist.gov", 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",