Add OneHot to the list of formulas. Fix #263.

getkeops · Aug 3, 2022 · 57bf8f6 · 57bf8f6
1 parent a884c71
commit 57bf8f6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keopscore/keopscore/formulas/maths/__init__.py b/keopscore/keopscore/formulas/maths/__init__.py
@@ -32,6 +32,7 @@
 from .Mult import Mult
 from .Norm2 import Norm2
 from .Normalize import Normalize
+from .OneHot import OneHot
 from .Pow import Pow
 from .Powf import Powf
 from .ReLU import ReLU
@@ -94,6 +95,7 @@
     "Mult",
     "Norm2",
     "Normalize",
+    "OneHot",
     "Pow",
     "Powf",
     "ReLU",