diff --git a/pyFTS/partitioners/partitioner.py b/pyFTS/partitioners/partitioner.py index 14eb4a8..aef4172 100644 --- 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: