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():
|