0.15.20

- xai.py determines shape of the dataset also for tensors
- save_experiment in spot uses copy if deep.copy fails

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotpython"
-version = "0.15.19"
+version = "0.15.20"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]

src/spotpython/plot/xai.py

@@ -618,7 +618,10 @@ def get_attributions(
     model = model.to("cpu")
     model.eval()
     dataset = fun_control["data_set"]
-    n_features = dataset.data.shape[1]
+    try:
+        n_features = dataset.data.shape[1]
+    except AttributeError:
+        n_features = dataset.tensors[0].shape[1]
     if feature_names is None:
         feature_names = [f"x{i}" for i in range(n_features)]
     batch_size = config["batch_size"]
@@ -745,7 +748,10 @@ def get_layer_conductance(spot_tuner, fun_control, layer_idx) -> np.ndarray:
     model.eval()
 
     dataset = fun_control["data_set"]
-    n_features = dataset.data.shape[1]
+    try:
+        n_features = dataset.data.shape[1]
+    except AttributeError:
+        n_features = dataset.tensors[0].shape[1]
     if feature_names is None:
         feature_names = [f"x{i}" for i in range(n_features)]
     batch_size = config["batch_size"]

src/spotpython/spot/spot.py

@@ -2314,7 +2314,7 @@ def save_experiment(self, filename=None) -> None:
         except Exception as e:
             logger.warning("Warning: Could not copy/save spot_tuner object!")
             logger.warning(f"Error: {e}")
-            spot_tuner = None
+            spot_tuner = copy(self)
         experiment = {
             "design_control": design_control,
             "fun_control": fun_control,