Minor improvement in unique

INGEOTEC · Apr 11, 2024 · 61b63f4 · 61b63f4
1 parent 1631f30
commit 61b63f4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/EvoMSA/__init__.py b/EvoMSA/__init__.py
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-__version__ = '2.0.8'
+__version__ = '2.0.9'
 
 try:
     from EvoMSA.text_repr import BoW, TextRepresentations, StackGeneralization, DenseBoW

diff --git a/EvoMSA/unique.py b/EvoMSA/unique.py
@@ -82,7 +82,7 @@ def unique_rest(frst, rest):
         transform = BoW(**bow_params).transform
     X = transform(D)
     init = 0
-    pool = np.arange(len(D))
+    pool = np.arange(X.shape[0])
     with tqdm(total=pool.shape[0]) as _tqdm:
         while init < pool.shape[0]:
             past = pool[:init]