3
5
Fork 0
mirror of git://git.savannah.gnu.org/guix.git synced 2023-12-14 03:33:07 +01:00
guix/gnu/packages/patches/python-keras-integration-test.patch
Marius Bakke b16e082799
gnu: python-keras: Add upstream patch.
* gnu/packages/patches/python-keras-integration-test.patch: New file.
* gnu/local.mk (dist_patch_DATA): Adjust accordingly.
* gnu/packages/machine-learning.scm (python-keras)[source](patches): New field.
2019-07-19 15:44:26 +02:00

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