Fix missing import

Signed-off-by: Emanuele Ballarin <[email protected]>
emaballarin · Jun 9, 2024 · 74e3037 · 74e3037
1 parent 32576a9
commit 74e3037
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/foveatorch/modfoveate.py b/foveatorch/modfoveate.py
@@ -259,7 +259,7 @@ def forward(self, xinput: Tensor, fixation_pts: Tensor | None = None) -> Tensor:
         )
 
 
-class FovealPyramidBlur(nn.Module):
+class FovealPyramidBlur(th.nn.Module):
     def __init__(
         self,
         pyramid_level: int,

diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@ def read(fname):
 
 setup(
     name=PACKAGENAME,
-    version="0.1.5",
+    version="0.1.6",
     author="Emanuele Ballarin",
     author_email="[email protected]",
     url="https://github.com/emaballarin/foveatorch",