diff --git a/genno/core/computer.py b/genno/core/computer.py index 531c634e..5607204f 100644 --- a/genno/core/computer.py +++ b/genno/core/computer.py @@ -546,9 +546,6 @@ def infer_keys( result = list(map(partial(self.graph.infer, dims=dims), keys)) - if None in result: - raise KeyError(keys[result.index(None)]) - return result[0] if single else tuple(result) def __contains__(self, item):