Cleanup/remove deprecated structures (#51)

* rm attention operators * rm deprecated deep o variants
JakobEliasWagner · Aug 18, 2024 · 7c5776a · 7c5776a
1 parent aa05765
commit 7c5776a
Show file tree

Hide file tree

Showing 12 changed files with 0 additions and 743 deletions.
diff --git a/src/nos/networks/attention/__init__.py b/src/nos/networks/attention/__init__.py
diff --git a/src/nos/networks/attention/function_encoder.py b/src/nos/networks/attention/function_encoder.py
diff --git a/src/nos/networks/attention/heterogeneous_normalized_attention.py b/src/nos/networks/attention/heterogeneous_normalized_attention.py
diff --git a/src/nos/networks/attention/multi_head_attention.py b/src/nos/networks/attention/multi_head_attention.py
diff --git a/src/nos/networks/attention/scaled_dot_product_attention.py b/src/nos/networks/attention/scaled_dot_product_attention.py
diff --git a/src/nos/operators/__init__.py b/src/nos/operators/__init__.py
@@ -1,8 +1,3 @@
-from .attention import (
-    AttentionOperator,
-    FunctionEncoder,
-    TransformerOperator,
-)
 from .belnet import (
     BelNet,
 )
@@ -12,12 +7,6 @@
 from .deep_neural_operator import (
     DeepNeuralOperator,
 )
-from .deep_o_branch_fno import (
-    DeepOBranchFNO,
-)
-from .deep_o_fno import (
-    DeepOFNO,
-)
 from .deep_o_net import (
     DeepONet,
 )
@@ -44,17 +33,12 @@
 
 __all__ = [
     "MeanStackNeuralOperator",
-    "DeepOBranchFNO",
     "DeepDotOperator",
     "NeuralOperator",
     "DeepNeuralOperator",
     "DeepONet",
     "FourierNeuralOperator",
     "DeepRootOperator",
-    "DeepOFNO",
-    "AttentionOperator",
-    "FunctionEncoder",
-    "TransformerOperator",
     "BelNet",
     "to_json",
     "to_pt",

diff --git a/src/nos/operators/attention/__init__.py b/src/nos/operators/attention/__init__.py