Fixes Issue #216 (#217)

* fixed missing imports to keep SavePath on the correct part of the API for backwards compat

* linted

spock/__init__.py

@@ -10,12 +10,13 @@
 """
 
 from spock._version import get_versions
+from spock.backend.typed import SavePath
 from spock.builder import ConfigArgBuilder
 from spock.config import spock
 
 SpockBuilder = ConfigArgBuilder
 
-__all__ = ["args", "builder", "config", "spock", "SpockBuilder"]
+__all__ = ["args", "builder", "config", "SavePath", "spock", "SpockBuilder"]
 
 __version__ = get_versions()["version"]
 del get_versions

spock/__init__.pyi

@@ -1,11 +1,13 @@
-from typing import Any, Callable, List, Optional, Tuple, Type, TypeVar, Union, overload
+from typing import Any, Callable, List, Optional, Tuple, TypeVar, Union, overload
 
 from attr import attrib, field
 
+from spock.backend.typed import SavePath
 from spock.builder import ConfigArgBuilder
 
 _T = TypeVar("_T")
 _C = TypeVar("_C", bound=type)
+SavePath: SavePath
 
 # Note: from here
 # https://github.com/python-attrs/attrs/blob/main/src/attr/__init__.pyi

spock/backend/config.pyi

@@ -4,6 +4,7 @@ from attr import attrib, field
 
 _T = TypeVar("_T")
 _C = TypeVar("_C", bound=type)
+SavePath: SavePath
 
 # Note: from here
 # https://github.com/python-attrs/attrs/blob/main/src/attr/__init__.pyi

spock/config.py

@@ -6,10 +6,12 @@
 """Creates the spock config decorator that wraps attrs"""
 
 from spock.backend.config import spock_attr
+from spock.backend.typed import SavePath
 from spock.utils import _is_spock_instance
 
 # Simplified decorator for attrs
 spock = spock_attr
+SavePath = SavePath
 
 # Public alias for checking if an object is a @spock annotated class
 isinstance_spock = _is_spock_instance