From a04b33199d3655157a9635de79c5244d703b0c03 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20tom=20W=C3=B6rden?= <h.tomwoerden@indiscale.com>
Date: Mon, 8 Jan 2024 10:48:05 +0100
Subject: [PATCH] DOC: add type hint
---
src/caoscrawler/scanner.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/caoscrawler/scanner.py b/src/caoscrawler/scanner.py
index 18d722ed..3f8b85f4 100644
--- a/src/caoscrawler/scanner.py
+++ b/src/caoscrawler/scanner.py
@@ -36,6 +36,7 @@ import importlib
import logging
import os
import warnings
+from collections.abc import Callable
from typing import Any, Optional, Type, Union
import caosdb as db
@@ -181,7 +182,7 @@ def create_converter_registry(definition: dict):
return converter_registry
-def create_transformer_registry(definition: dict):
+def create_transformer_registry(definition: dict[str, dict[str, str]]):
"""
Currently the transformer registry is a dictionary containing for each transformer:
- key is the short code, abbreviation for the converter class name
@@ -201,7 +202,7 @@ def create_transformer_registry(definition: dict):
with open(str(files('caoscrawler').joinpath('default_transformers.yml')), "r") as f:
transformer_def: dict[str, dict[str, str]] = yaml.safe_load(f)
- registry = {}
+ registry: dict[str, Callable[[Any, dict], Any]] = {}
# More transformers from definition file:
if "Transformers" in definition:
for key, entry in definition["Transformers"].items():
--
GitLab