remove attn cnp

remove attn cnp remove attn cnp tests
alan-turing-institute · Aug 12, 2024 · f289db8 · f289db8
1 parent 6d417f3
commit f289db8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 328 deletions.
diff --git a/autoemulate/emulators/conditional_neural_process.py b/autoemulate/emulators/conditional_neural_process.py
@@ -1,3 +1,5 @@
+import warnings
+
 import numpy as np
 import torch
 from scipy.stats import loguniform
@@ -14,7 +16,6 @@
 from skorch.callbacks import LRScheduler
 from torch import nn
 
-from autoemulate.emulators.neural_networks.attn_cnp_module import AttnCNPModule
 from autoemulate.emulators.neural_networks.cnp_module import CNPModule
 from autoemulate.emulators.neural_networks.datasets import cnp_collate_fn
 from autoemulate.emulators.neural_networks.datasets import CNPDataset
@@ -136,6 +137,9 @@ def __init__(
         self.activation = activation
         self.optimizer = optimizer
         self.normalize_y = normalize_y
+        if attention:
+            warnings.warn("Attention is not implemented yet, setting to False.")
+            attention = False
         self.attention = attention
         self.device = device
         self.random_state = random_state

diff --git a/autoemulate/emulators/neural_networks/attn_cnp_module.py b/autoemulate/emulators/neural_networks/attn_cnp_module.py
diff --git a/tests/test_attn_cnp.py b/tests/test_attn_cnp.py