adding a check for the ndarray data type

2024-08-14 11:55:54 +04:00 · 2024-08-14 11:55:54 +04:00 · ccc0196f56
commit ccc0196f56
parent 3ec1b4c5ce
1 changed files with 1 additions and 1 deletions
--- a/pyFTS/partitioners/partitioner.py
+++ b/pyFTS/partitioners/partitioner.py
@ -47,7 +47,7 @@ class Partitioner(object):

            data = kwargs.get('data',[None])
            
-            if len(data.shape) > 1:
+            if isinstance(data, np.ndarray) and len(data.shape) > 1:
                warnings.warn(f"An ndarray of dimension greater than 1 is used. shape.len(): {len(data.shape)}")

            if self.indexer is not None: