Merge pull request #18 from IBM/pre-v.0.3.3

Pre v.0.3.3
IBM · Aug 15, 2024 · 0c4dc95 · 0c4dc95
2 parents 6d7e374 + bbbe684
commit 0c4dc95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/autopeptideml/utils/embeddings.py b/autopeptideml/utils/embeddings.py
@@ -69,7 +69,7 @@ def compute_batch(self, batch: list, average_pooling: bool):
                                       decoder_input_ids=inputs['input_ids']
                                       ).last_hidden_state
             else:
-                embd_rpr = self.model(**input_ids)
+                embd_rpr = self.model(**inputs)
 
         output = []
         for idx in range(len(batch)):

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