Fix write_file destination parameter type hint (#430)

oceanprotocol · Aug 6, 2021 · 3df5238 · 3df5238
1 parent 5a4cdb6
commit 3df5238
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ocean_lib/data_provider/data_service_provider.py b/ocean_lib/data_provider/data_service_provider.py
@@ -24,7 +24,6 @@
 from ocean_lib.ocean.env_constants import ENV_PROVIDER_API_VERSION
 from ocean_lib.web3_internal.transactions import sign_hash
 from ocean_lib.web3_internal.wallet import Wallet
-from py._path.local import LocalPath
 from requests.exceptions import InvalidURL
 from requests.models import PreparedRequest, Response
 from requests.sessions import Session
@@ -530,7 +529,9 @@ def build_fileinfo(provider_uri: str) -> Tuple[str, str]:
 
     @staticmethod
     def write_file(
-        response: Response, destination_folder: Union[LocalPath, str], file_name: str
+        response: Response,
+        destination_folder: Union[str, bytes, os.PathLike],
+        file_name: str,
     ) -> None:
         """
         Write the response content in a file in the destination folder.