fix old lstnet import and name

Author: LarsKue

bayesflow/__init__.py

@@ -36,7 +36,7 @@ def setup():
         # turn off gradients by default
         import torch
 
-        torch.autograd.set_grad_enabled(False)
+        # torch.autograd.set_grad_enabled(False)
 
         logging.warning("Disabling gradients by default. Use\nwith torch.enable_grad():\nin custom training loops.")
 

bayesflow/networks/__init__.py

@@ -5,7 +5,7 @@
 from .inference_network import InferenceNetwork
 from .point_inference_network import PointInferenceNetwork
 from .mlp import MLP
-from .lstnet import TimeSeriesNetwork
+from .time_series_network import TimeSeriesNetwork
 from .summary_network import SummaryNetwork
 from .transformers import SetTransformer, TimeSeriesTransformer, FusionTransformer
 

bayesflow/networks/lstnet/__init__.py renamed to bayesflow/networks/time_series_network/__init__.py

@@ -36,7 +36,7 @@ def find_summary_network(summary_network: SummaryNetwork | str, **kwargs) -> Sum
             return bayesflow.networks.FusionTransformer(**kwargs)
         case "time_series_transformer":
             return bayesflow.networks.TimeSeriesTransformer(**kwargs)
-        case "lstnet":
+        case "time_series_network":
             return bayesflow.networks.LSTNet(**kwargs)
         case str() as unknown_network:
             raise ValueError(f"Unknown summary network: '{unknown_network}'")