diff --git a/src/serve.py b/src/serve.py index 8b76724..1948abf 100644 --- a/src/serve.py +++ b/src/serve.py @@ -31,6 +31,8 @@ def make_callback(analysis_endpoint): operations = message.get("operations", ["/"]) results = map(perform_operation, operations) result = dict(zip(operations, results)) + if list(result.keys()) == ["/"]: + result = list(result.values())[0] return result return callback