mirror of
https://git.in.rschanz.org/ryan77627/guix.git
synced 2024-12-26 06:18:07 -05:00
20 lines
799 B
Diff
20 lines
799 B
Diff
|
Fix a test failure with recent versions of ... Pytest? Python?
|
||
|
|
||
|
Taken from upstream:
|
||
|
https://github.com/keras-team/keras/commit/fe35050a8f18dc52304aa8da4e463eececa25240
|
||
|
|
||
|
diff --git a/tests/integration_tests/applications_test.py b/tests/integration_tests/applications_test.py
|
||
|
index 979f2f2abd6..6e3b57fa8e2 100644
|
||
|
--- a/tests/integration_tests/applications_test.py
|
||
|
+++ b/tests/integration_tests/applications_test.py
|
||
|
@@ -58,7 +58,8 @@ def _test_application_basic(app, last_dim=1000):
|
||
|
def _test_application_notop(app, last_dim):
|
||
|
output_shape = _get_output_shape(
|
||
|
lambda: app(weights=None, include_top=False))
|
||
|
- assert output_shape == (None, None, None, last_dim)
|
||
|
+ assert len(output_shape) == 4
|
||
|
+ assert output_shape[-1] == last_dim
|
||
|
|
||
|
|
||
|
def test_mobilenet_v2_legacy_import():
|