From 28332842391024092905df2d57e5858dbde08b9c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mose=20M=C3=BCller?= Date: Mon, 6 Nov 2023 10:06:08 +0100 Subject: [PATCH] chore: updating types, removes unused imports --- src/pydase/data_service/abstract_data_service.py | 3 ++- src/pydase/data_service/data_service.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/pydase/data_service/abstract_data_service.py b/src/pydase/data_service/abstract_data_service.py index 9264078..f84b29e 100644 --- a/src/pydase/data_service/abstract_data_service.py +++ b/src/pydase/data_service/abstract_data_service.py @@ -1,6 +1,7 @@ from __future__ import annotations from abc import ABC +from pathlib import Path from typing import TYPE_CHECKING, Any, Optional if TYPE_CHECKING: @@ -14,4 +15,4 @@ class AbstractDataService(ABC): _task_manager: TaskManager _callback_manager: CallbackManager _autostart_tasks: dict[str, tuple[Any]] - _filename: Optional[str] + _filename: Optional[str | Path] diff --git a/src/pydase/data_service/data_service.py b/src/pydase/data_service/data_service.py index d3c4c61..a51aa97 100644 --- a/src/pydase/data_service/data_service.py +++ b/src/pydase/data_service/data_service.py @@ -1,6 +1,7 @@ import logging import warnings from enum import Enum +from pathlib import Path from typing import Any, Optional, get_type_hints import rpyc @@ -8,7 +9,6 @@ import rpyc import pydase.units as u from pydase.data_service.abstract_data_service import AbstractDataService from pydase.data_service.callback_manager import CallbackManager -from pydase.data_service.state_manager import StateManager from pydase.data_service.task_manager import TaskManager from pydase.utils.helpers import ( convert_arguments_to_hinted_types, @@ -41,7 +41,7 @@ def process_callable_attribute(attr: Any, args: dict[str, Any]) -> Any: class DataService(rpyc.Service, AbstractDataService): def __init__(self, **kwargs: Any) -> None: - self._filename: Optional[str] = None + self._filename: Optional[str | Path] = None self._callback_manager: CallbackManager = CallbackManager(self) self._task_manager = TaskManager(self)