diff --git a/src/TorchSisso/model.py b/src/TorchSisso/model.py index de27311..010be1a 100644 --- a/src/TorchSisso/model.py +++ b/src/TorchSisso/model.py @@ -56,7 +56,7 @@ def __init__(self,data,operators=None,multi_task = None,n_expansion=None,n_term= else: self.dimension = n_term - else: self.sis_features = k + self.sis_features = k self.relational_units = relational_units