Fixed some dependency issues

fractalis/analytics/task.py

@@ -8,7 +8,7 @@ from typing import List, Tuple
 
 from pandas import read_csv, DataFrame
 from celery import Task
-from Crypto.Cipher import AES
+from Cryptodome.Cipher import AES
 
 from fractalis import redis, app
 from fractalis.utils import get_cache_encrypt_key

fractalis/data/etl.py

@@ -5,7 +5,7 @@ import json
 import logging
 import os
 
-from Crypto.Cipher import AES
+from Cryptodome.Cipher import AES
 from celery import Task
 from pandas import DataFrame
 

fractalis/data/etls/transmart/etl_highdim.py

@@ -6,7 +6,6 @@ import requests
 from pandas import DataFrame
 
 from fractalis.data.etl import ETL
-from fractalis.data.etls.transmart import observations_pb2
 
 logger = logging.getLogger(__name__)
 

fractalis/data/integrity/numerical_array.py

@@ -39,7 +39,7 @@ class NumericalArrayIntegrityCheck(IntegrityCheck):
             error = "'value' column must be of type 'np.int' or 'np.float'."
             logger.error(error)
             raise ValueError(error)
-        if len(pd.unique(data[['id', 'feature']].values)) != data.shape[0]:
+        if data.groupby(['id', 'feature']).count().shape[0] != data.shape[0]:
             error = "Every combination of 'id' and 'feature' must be unique."
             logger.error(error)
             raise ValueError(error)
\ No newline at end of file

setup.py

@@ -22,7 +22,6 @@ setup(
         'sklearn',
         'requests',
         'PyYAML',
-        'pycrypto',
         'pycryptodomex',
         'rpy2'
     ],