fix error in flatten protVec

anazhmetdin · Dec 9, 2020 · 2dbca6f · 2dbca6f
1 parent f3a1a20
commit 2dbca6f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/protencoder/protVec.py b/protencoder/protVec.py
@@ -1,5 +1,6 @@
 import biovec
 import random
+import numpy as np
 from protencoder.protencoder import encoder
 
 
@@ -38,7 +39,10 @@ def encode(self):
                 seq.replace("O", "k")
             encoded = self.pv.to_vecs(seq)
             if self.flatten:
-                encoded = encoded.flatten()
+                a = np.array([])
+                for i in encoded:
+                    a = np.concatenate((a, i), axis=None)
+                encoded = a
             self.handler.seqDict[prot] = encoded
 
     def read(self, seqPath):

diff --git a/setup.py b/setup.py
@@ -49,7 +49,7 @@
     test_suite='tests',
     tests_require=test_requirements,
     url='https://github.com/anegm98/protencoder',
-    version='1.4.3',
+    version='1.4.4',
     zip_safe=False,
     package_data={'protencoder': ['data/*']}
 )