From 02d523f0ee6b6a48eba1cca39937f084cea6b0e0 Mon Sep 17 00:00:00 2001 From: Kien Dang Date: Tue, 28 Nov 2023 00:37:53 +0800 Subject: [PATCH] Minor typing improvement --- packages/syft/src/syft/custom_worker/config.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/syft/src/syft/custom_worker/config.py b/packages/syft/src/syft/custom_worker/config.py index a2f9115d65b..d6244585c83 100644 --- a/packages/syft/src/syft/custom_worker/config.py +++ b/packages/syft/src/syft/custom_worker/config.py @@ -1,6 +1,8 @@ # stdlib from hashlib import sha256 from pathlib import Path +from typing import Any +from typing import Dict from typing import List from typing import Union @@ -77,7 +79,7 @@ class CustomWorkerConfig(SyftBaseModel): version: str = "1" @classmethod - def from_dict(cls, config: dict) -> Self: + def from_dict(cls, config: Dict[str, Any]) -> Self: return cls(**config) @classmethod