1
- import platform
2
-
3
1
import numpy
4
2
import pytest
5
3
from hypothesis import given , settings
@@ -36,7 +34,7 @@ def test_linear_dimensions_on_data():
36
34
y .max .assert_called_with ()
37
35
38
36
39
- @pytest .mark .skipif ( platform . machine () == "win_amd64" , reason = "Flaky, skip temporarily" )
37
+ @pytest .mark .skip ( reason = "Flaky, skip temporarily" )
40
38
@given (arrays_OI_O_BI (max_batch = 8 , max_out = 8 , max_in = 8 ))
41
39
def test_begin_update_matches_predict (W_b_input ):
42
40
model = get_model (W_b_input )
@@ -47,7 +45,7 @@ def test_begin_update_matches_predict(W_b_input):
47
45
assert_allclose (fwd_via_begin_update , fwd_via_predict_batch )
48
46
49
47
50
- @pytest .mark .skipif ( platform . machine () == "win_amd64" , reason = "Flaky, skip temporarily" )
48
+ @pytest .mark .skip ( reason = "Flaky, skip temporarily" )
51
49
@given (arrays_OI_O_BI (max_batch = 8 , max_out = 8 , max_in = 8 ))
52
50
def test_finish_update_calls_optimizer_with_weights (W_b_input ):
53
51
model = get_model (W_b_input )
@@ -69,7 +67,7 @@ def sgd(key, data, gradient, **kwargs):
69
67
assert (model .id , name ) in seen_keys
70
68
71
69
72
- @pytest .mark .skipif ( platform . machine () == "win_amd64" , reason = "Flaky, skip temporarily" )
70
+ @pytest .mark .skip ( reason = "Flaky, skip temporarily" )
73
71
@settings (max_examples = 100 )
74
72
@given (arrays_OI_O_BI (max_batch = 8 , max_out = 8 , max_in = 8 ))
75
73
def test_predict_small (W_b_input ):
@@ -92,7 +90,7 @@ def test_predict_small(W_b_input):
92
90
assert_allclose (predicted_output , expected_output , rtol = 0.01 , atol = 0.01 )
93
91
94
92
95
- @pytest .mark .skipif ( platform . machine () == "win_amd64" , reason = "Flaky, skip temporarily" )
93
+ @pytest .mark .skip ( reason = "Flaky, skip temporarily" )
96
94
@given (arrays_OI_O_BI (max_batch = 20 , max_out = 30 , max_in = 30 ))
97
95
@settings (deadline = None )
98
96
def test_predict_extensive (W_b_input ):
@@ -115,7 +113,7 @@ def test_predict_extensive(W_b_input):
115
113
assert_allclose (predicted_output , expected_output , rtol = 1e-04 , atol = 0.0001 )
116
114
117
115
118
- @pytest .mark .skipif ( platform . machine () == "win_amd64" , reason = "Flaky, skip temporarily" )
116
+ @pytest .mark .skip ( reason = "Flaky, skip temporarily" )
119
117
@given (arrays_OI_O_BI (max_batch = 8 , max_out = 8 , max_in = 8 ))
120
118
def test_dropout_gives_zero_activations (W_b_input ):
121
119
model = chain (get_model (W_b_input ), Dropout (1.0 ))
@@ -125,7 +123,7 @@ def test_dropout_gives_zero_activations(W_b_input):
125
123
assert all (val == 0.0 for val in fwd_dropped .flatten ())
126
124
127
125
128
- @pytest .mark .skipif ( platform . machine () == "win_amd64" , reason = "Flaky, skip temporarily" )
126
+ @pytest .mark .skip ( reason = "Flaky, skip temporarily" )
129
127
@given (arrays_OI_O_BI (max_batch = 8 , max_out = 8 , max_in = 8 ))
130
128
def test_dropout_gives_zero_gradients (W_b_input ):
131
129
model = chain (get_model (W_b_input ), Dropout (1.0 ))
0 commit comments