oops

srigas · Jun 11, 2024 · be15844 · be15844
1 parent e011116
commit be15844
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/jaxkan/models/KAN.py b/jaxkan/models/KAN.py
@@ -5,7 +5,7 @@
 from flax.linen import initializers
 from flax.core import unfreeze
 
-from models.KANLayer import KANLayer
+from .KANLayer import KANLayer
 
 
 class KAN(nn.Module):

diff --git a/jaxkan/models/KANLayer.py b/jaxkan/models/KANLayer.py
@@ -3,8 +3,8 @@
 from flax import linen as nn
 from flax.linen import initializers
 
-from bases.splines import get_spline_basis
-from utils.general import solve_full_lstsq
+from ..bases.splines import get_spline_basis
+from ..utils.general import solve_full_lstsq
 
 
 class KANLayer(nn.Module):

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='jaxkan',
-    version='0.1.1',
+    version='0.1.3',
     packages=find_packages(),
     install_requires=[
         'numpy==1.26.4',