Skip to content

Commit

Permalink
ran make fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
shichengzhou-db committed Sep 5, 2024
1 parent 49a7f78 commit 34fe6bf
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
4 changes: 4 additions & 0 deletions databricks/sdk/mixins/files.py
Original file line number Diff line number Diff line change
Expand Up @@ -264,6 +264,7 @@ def __exit__(self, __t, __value, __traceback):
def __repr__(self) -> str:
return f"<_VolumesIO {self._path} {'read' if self.readable() else 'write'}=True>"


class _ModelsIO(BinaryIO):

def __init__(self, api: files.FilesAPI, path: str, *, read: bool, write: bool, overwrite: bool):
Expand Down Expand Up @@ -361,6 +362,7 @@ def __exit__(self, __t, __value, __traceback):
def __repr__(self) -> str:
return f"<_ModelsIO {self._path} {'read' if self.readable() else 'write'}=True>"


class _Path(ABC):

@abstractmethod
Expand Down Expand Up @@ -562,6 +564,7 @@ def delete(self, *, recursive=False):
def __repr__(self) -> str:
return f'<_VolumesPath {self._path}>'


class _ModelsPath(_Path):

def __init__(self, api: files.FilesAPI, src: Union[str, pathlib.Path]):
Expand Down Expand Up @@ -630,6 +633,7 @@ def delete(self, *, recursive=False):
def __repr__(self) -> str:
return f'<_ModelsPath {self._path}>'


class _DbfsPath(_Path):

def __init__(self, api: files.DbfsAPI, src: str):
Expand Down
2 changes: 1 addition & 1 deletion tests/test_dbfs_mixins.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

from databricks.sdk.errors import NotFound
from databricks.sdk.mixins.files import (DbfsExt, _DbfsPath, _LocalPath,
_VolumesPath, _ModelsPath)
_ModelsPath, _VolumesPath)


def test_moving_dbfs_file_to_local_dir(config, tmp_path, mocker):
Expand Down

0 comments on commit 34fe6bf

Please sign in to comment.