Skip to content

Commit c105b1d

Browse files
committed
add is_multioutput method to DummyModel in tests
1 parent 5136c73 commit c105b1d

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

tests/experimental/test_experimental_base.py

+4
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,10 @@ def get_tune_config():
119119
"batch_size": [16],
120120
}
121121

122+
@staticmethod
123+
def is_multioutput():
124+
return False
125+
122126
def setup_method(self):
123127
"""
124128
Define the PyTorchBackend instance.

tests/experimental/test_experimental_model_selection.py

+4
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,10 @@ def predict(self, x):
2525
def get_tune_config():
2626
return {}
2727

28+
@staticmethod
29+
def is_multioutput():
30+
return False
31+
2832
x = torch.tensor(np.arange(32)).float()
2933
y = 2 * x
3034
dataset = TensorDataset(x, y)

0 commit comments

Comments
 (0)