Merge pull request #12 from usnistgov/develop

Fix data module for classification models.
usnistgov · May 8, 2021 · 6d05251 · 6d05251
2 parents efcc0ab + 4853944
commit 6d05251
Show file tree

Hide file tree

Showing 3 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__ = "2021.5.7"
+__version__ = "2021.5.8"
diff --git a/alignn/data.py b/alignn/data.py
@@ -277,7 +277,7 @@ def get_train_val_loaders(
                             i[target],
                             type(i[target]),
                         )
-            dat.append(i)
+                dat.append(i)
 
         # id_test = ids[-test_size:]
         # if standardize:

diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setuptools.setup(
     name="alignn",  # Replace with your own username
-    version="2021.5.7",
+    version="2021.5.8",
     author="Kamal Choudhary, Brian DeCost",
     author_email="[email protected]",
     description="alignn",