From 7865a767c7343e11d2e5f75c1446a81221e81f7f Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Wed, 15 Jun 2022 14:00:09 +0200 Subject: [PATCH] added type hint --- pyinfra/visitor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyinfra/visitor.py b/pyinfra/visitor.py index 55a8a87..155d5bd 100644 --- a/pyinfra/visitor.py +++ b/pyinfra/visitor.py @@ -58,7 +58,7 @@ class StorageStrategy(ResponseStrategy): def __init__(self, storage): self.storage = storage - def handle_response(self, body): + def handle_response(self, body: dict): response_object_descriptor = self.get_response_object_descriptor(body) self.storage.put_object(**response_object_descriptor, data=gzip.compress(json.dumps(body).encode())) body.pop("data")