lint

python_modules/libraries/dagster-openai/dagster_openai/resources.py

@@ -222,10 +222,7 @@ def setup_for_execution(self, context: InitResourceContext) -> None:
  kwargs["project"] = self.project
  if self.base_url:
  kwargs["base_url"] = self.base_url
- self._client = Client(
- api_key=self.api_key,
- **kwargs
- )
+ self._client = Client(api_key=self.api_key, **kwargs)
 
  @public
  @contextmanager

python_modules/libraries/dagster-openai/dagster_openai_tests/test_resources.py

@@ -1,4 +1,3 @@
-from mock.mock import Base
 import pytest
 from dagster import (
  AssetExecutionContext,
@@ -20,7 +19,6 @@
 from dagster._utils.test import wrap_op_in_graph_and_execute
 from dagster_openai import OpenAIResource, with_usage_metadata
 from mock import ANY, MagicMock, patch
-from openai import OpenAI
 
 
 @patch("dagster_openai.resources.Client")
@@ -39,7 +37,7 @@ def test_openai_client_with_config(mock_client) -> None:
  api_key="xoxp-1234123412341234-12341234-1234",
  organization="foo",
  project="bar",
- base_url="https://foo.bar"
+ base_url="https://foo.bar",
  )
  openai_resource.setup_for_execution(build_init_resource_context())
 
@@ -49,7 +47,7 @@ def test_openai_client_with_config(mock_client) -> None:
  api_key="xoxp-1234123412341234-12341234-1234",
  organization="foo",
  project="bar",
- base_url="https://foo.bar"
+ base_url="https://foo.bar",
  )