diff --git a/python/docs/create_api_rst.py b/python/docs/create_api_rst.py index 253352767..7a2f75260 100644 --- a/python/docs/create_api_rst.py +++ b/python/docs/create_api_rst.py @@ -105,7 +105,9 @@ def _load_module_members(module_path: str, namespace: str) -> ModuleMembers: else ( "enum" if issubclass(type_, Enum) - else "Pydantic" if issubclass(type_, BaseModel) else "Regular" + else "Pydantic" + if issubclass(type_, BaseModel) + else "Regular" ) ) if hasattr(type_, "__slots__"): diff --git a/python/langsmith/__init__.py b/python/langsmith/__init__.py index ceb8bb56b..f3a1de90a 100644 --- a/python/langsmith/__init__.py +++ b/python/langsmith/__init__.py @@ -2,8 +2,6 @@ from typing import TYPE_CHECKING, Any -import parametrize as parametrize - if TYPE_CHECKING: from langsmith._expect import expect from langsmith._testing import test, unit