Merge pull request #21 from IBM/code-dev

🪲 Bug fix: Compatibility with Hestia 0.0.19 fixed
IBM · Sep 11, 2024 · 4125c1c · 4125c1c
2 parents e76cee0 + 13d2d7c
commit 4125c1c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/autopeptideml/autopeptideml.py b/autopeptideml/autopeptideml.py
@@ -473,7 +473,7 @@ def train_test_partition(
 
         df = df.sample(len(df), random_state=self.seed).reset_index(drop=True)
 
-        train_idx, test_idx = ccpart(
+        train_idx, test_idx, label_ids = ccpart(
             df=df,
             similarity_metric=alignment,
             field_name='sequence',

diff --git a/setup.py b/setup.py
@@ -52,6 +52,6 @@
     name='autopeptideml',
     packages=find_packages(exclude=['examples']),
     url='https://ibm.github.io/AutoPeptideML/',
-    version='1.0.0',
+    version='1.0.1',
     zip_safe=False,
 )