python3Packages.runway-python: fix build, enable tests

This commit is contained in:
Martin Weinelt 2021-05-07 20:25:29 +02:00 committed by Jonathan Ringer
parent 5ae90276c3
commit 12493f272b

View file

@ -1,33 +1,43 @@
{ lib
, buildPythonPackage
, fetchPypi
, pythonAtLeast
, fetchFromGitHub
, colorcet
, cryptography
, flask
, flask-compress
, flask-cors
, flask-sockets
, gevent
, imageio
, numpy
, scipy
, pillow
, gevent
, wget
, pyopenssl
, scipy
, six
, colorcet
, unidecode
, urllib3
, wget
, deepdiff
, pytestCheckHook
, pytestcov
, websocket_client
}:
buildPythonPackage rec {
pname = "runway-python";
version = "0.6.1";
src = fetchPypi {
inherit pname version;
sha256 = "66cf1517dd817bf6db3792608920274f964dd0ced8dabecd925b8bc17aa95740";
src = fetchFromGitHub {
owner = "runwayml";
repo = "model-sdk";
rev = version;
sha256 = "1ww2wai1qnly8i7g42vhkkbs4yp7wi9x4fjdxsg9fl3izjra0zs2";
};
propagatedBuildInputs = [
colorcet
cryptography
flask
flask-compress
flask-cors
@ -36,6 +46,7 @@ buildPythonPackage rec {
imageio
numpy
pillow
pyopenssl
scipy
six
unidecode
@ -43,13 +54,30 @@ buildPythonPackage rec {
wget
];
# tests are not packaged in the released tarball
doCheck = false;
pythonImportsCheck = [
"runway"
];
checkInputs = [
deepdiff
pytestCheckHook
pytestcov
websocket_client
];
disabledTests = [
# these tests require network
"test_file_deserialization_remote"
"test_file_deserialization_absolute_directory"
"test_file_deserialization_remote_directory"
] ++ lib.optionals (pythonAtLeast "3.9") [
# AttributeError: module 'base64' has no attribute 'decodestring
# https://github.com/runwayml/model-sdk/issues/99
"test_image_serialize_and_deserialize"
"test_segmentation_serialize_and_deserialize_colormap"
"test_segmentation_serialize_and_deserialize_labelmap"
];
meta = {
description = "Helper library for creating Runway models";
homepage = "https://github.com/runwayml/model-sdk";