add returns an AbstractSDRAM

SpiNNakerManchester · Oct 21, 2024 · f5946ca · f5946ca
1 parent 7666539
commit f5946ca
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/pacman/model/resources/constant_sdram.py b/pacman/model/resources/constant_sdram.py
@@ -54,7 +54,7 @@ def __eq__(self, other: Any) -> bool:
             return False
         return other.fixed == self.fixed
 
-    def __add__(self, other):
+    def __add__(self, other: AbstractSDRAM) -> AbstractSDRAM:
         if isinstance(other, ConstantSDRAM):
             return ConstantSDRAM(
                 self._sdram + other.fixed)

diff --git a/pacman/model/resources/shared_sdram.py b/pacman/model/resources/shared_sdram.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import math
-from typing import Any, Dict, Optional, Self, TextIO, Union
+from typing import Any, Dict, Optional, TextIO, Union
 
 import numpy
 
@@ -96,7 +96,7 @@ def __eq__(self, other: Any) -> bool:
         return self._shared == other._shared
 
     @overrides(AbstractSDRAM.__add__)
-    def __add__(self, other: AbstractSDRAM) -> Self:
+    def __add__(self, other: AbstractSDRAM) -> AbstractSDRAM:
         if isinstance(other, (ConstantSDRAM, VariableSDRAM)):
             return SharedSDRAM(self._shared, self._per_core + other)
         elif isinstance(other, SharedSDRAM):

diff --git a/pacman/model/resources/variable_sdram.py b/pacman/model/resources/variable_sdram.py
@@ -87,7 +87,7 @@ def __eq__(self, other: Any) -> bool:
         else:
             return False
 
-    def __add__(self, other: AbstractSDRAM) -> 'VariableSDRAM':
+    def __add__(self, other: AbstractSDRAM) -> AbstractSDRAM:
         if isinstance(other, ConstantSDRAM):
             return VariableSDRAM(
                 self._fixed_sdram + other.fixed,  self._per_timestep_sdram)