concepts: introduce the Craft method

Signed-off-by: Frederic Boisnard <frederic.boisnard@irt-saintexupery.com>
deel-ai · Oct 9, 2023 · 3327cac · 3327cac
1 parent 5bfb885
commit 3327cac
Show file tree

Hide file tree

Showing 5 changed files with 1,505 additions and 6 deletions.
diff --git a/xplique/concepts/__init__.py b/xplique/concepts/__init__.py
@@ -1,6 +1,13 @@
-"""
-Concept based methods
-"""
-
-from .cav import Cav
-from .tcav import Tcav
+"""
+Concept based methods
+"""
+
+from .cav import Cav
+from .tcav import Tcav
+from .craft import BaseCraft, DisplayImportancesOrder
+from .craft_manager import BaseCraftManager
+from .craft_tf import CraftTf, CraftManagerTf
+try:
+    from .craft_torch import CraftTorch, CraftManagerTorch
+except ImportError:
+    pass