Commit ebb30130 authored by Mitar's avatar Mitar
Browse files

Improved check.

parent 0c6be646
Pipeline #23460433 passed with stage
in 5 minutes and 10 seconds
......@@ -102,7 +102,7 @@ class SumPrimitive(transformer.TransformerPrimitiveBase[Inputs, Outputs, Hyperpa
super().__init__(hyperparams=hyperparams, docker_containers=docker_containers)
# We cannot check for expected ports here because during class construction, a mock value is passed which has empty ports dict.
if DOCKER_KEY not in self.docker_containers:
if not self.docker_containers or DOCKER_KEY not in self.docker_containers:
raise ValueError("Docker key '{docker_key}' missing among provided Docker containers.".format(docker_key=DOCKER_KEY))
def _convert_value(self, value: typing.Any) -> typing.Union[numpy.ndarray, typing.List, typing.Any]:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment