1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
Index: skorch/skorch/tests/test_net.py
===================================================================
--- skorch.orig/skorch/tests/test_net.py
+++ skorch/skorch/tests/test_net.py
@@ -421,6 +421,7 @@ class TestNeuralNet:
return net_pickleable
+ @pytest.mark.skip()
@pytest.fixture
def pickled_cuda_net_path(self, net_pickleable, data):
path = os.path.join('skorch', 'tests', 'net_cuda.pkl')
@@ -437,6 +438,7 @@ class TestNeuralNet:
pickle.dump(net_pickleable, f)
return path
+ @pytest.mark.skip()
@pytest.mark.parametrize('cuda_available', {False, torch.cuda.is_available()})
def test_pickle_load(self, cuda_available, pickled_cuda_net_path):
with patch('torch.cuda.is_available', lambda *_: cuda_available):
@@ -1050,6 +1052,7 @@ class TestNeuralNet:
net.load_params(**kwargs)
net.predict(data[0]) # does not raise
+ @pytest.mark.skip()
@pytest.mark.parametrize('parameter,name', [
('f_params', 'net_cuda.pt'),
('f_optimizer', 'optimizer_cuda.pt'),
|
