add aliasing kan=multkan

KindXiaoming · Jul 8, 2024 · 3c71198 · 3c71198
1 parent 54d2758
commit 3c71198
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/kan/MultKAN.py b/kan/MultKAN.py
@@ -1454,4 +1454,7 @@ def module(self, start_layer, chain):
             self.symbolic_fun[cl].mask.data[np.ix_(id_out,id_in_other)] = 0.
             self.symbolic_fun[cl].mask.data[np.ix_(id_out_other,id_in)] = 0.
 
-        self.log_history('module')
+        self.log_history('module')
+
+
+KAN = MultKAN
diff --git a/kan/__init__.py b/kan/__init__.py
@@ -1,2 +1 @@
 from .MultKAN import *
-from .KAN import *