Formatted

famura · Jul 17, 2024 · e19eb60 · e19eb60
1 parent 51d2b2f
commit e19eb60
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/neuralfields/simple_neural_fields.py b/neuralfields/simple_neural_fields.py
@@ -369,14 +369,10 @@ def _init_capacity_heuristic(self, activation_nonlin: ActivationFunction) -> tor
         """
         if activation_nonlin is torch.sigmoid:
             # sigmoid(7.) approx 0.999
-            return PotentialBased.transform_to_opt_space(
-                torch.tensor([7.0], dtype=torch.get_default_dtype())
-            )
+            return PotentialBased.transform_to_opt_space(torch.tensor([7.0], dtype=torch.get_default_dtype()))
         elif activation_nonlin is torch.tanh:
             # tanh(3.8) approx 0.999
-            return PotentialBased.transform_to_opt_space(
-                torch.tensor([3.8], dtype=torch.get_default_dtype())
-            )
+            return PotentialBased.transform_to_opt_space(torch.tensor([3.8], dtype=torch.get_default_dtype()))
         raise NotImplementedError(
             "For the potential dynamics including a capacity, the initialization heuristic only supports "
             "the activation functions `torch.sigmoid` and `torch.tanh`!"