hopefully fixes everything, ahain now

Safe-DS · Jul 12, 2024 · 8fc9357 · 8fc9357
1 parent 790ed89
commit 8fc9357
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/src/safeds/ml/nn/layers/_convolutional2d_layer.py b/src/safeds/ml/nn/layers/_convolutional2d_layer.py
@@ -5,14 +5,13 @@
 from typing import TYPE_CHECKING, Any, Literal
 
 from safeds._utils import _structural_hash
-from safeds.ml.nn.typing import TensorShape
 
 from ._layer import Layer
 
 if TYPE_CHECKING:
     from torch import nn
 
-    from safeds.ml.nn.typing import ModelImageSize
+    from safeds.ml.nn.typing import ModelImageSize, TensorShape
 
 
 class Convolutional2DLayer(Layer):

diff --git a/src/safeds/ml/nn/layers/_pooling2d_layer.py b/src/safeds/ml/nn/layers/_pooling2d_layer.py
@@ -134,7 +134,7 @@ def __sizeof__(self) -> int:
             + sys.getsizeof(self._padding)
         )
 
-    def get_parameter_count(self, input_size: TensorShape) -> int:
+    def get_parameter_count(self, input_size: TensorShape) -> int:  # noqa: ARG002
         return 0
 
 

diff --git a/src/safeds/ml/nn/typing/_tensor_shape.py b/src/safeds/ml/nn/typing/_tensor_shape.py
@@ -41,7 +41,7 @@ def get_size(self, dimension: int | None = None) -> int:
             #TODO maybe add error message indicating that the dimension is out of range
             return 0        
         if(dimension is None):
-            return self._dims
+            return self._dims[0]
         return self._dims[dimension]
 
     def __hash__(self) -> int: