From c88a9988e385b15ad6ed6d3e2348262659e707f6 Mon Sep 17 00:00:00 2001 From: Sammy Sidhu Date: Sun, 24 Sep 2023 16:06:25 -0700 Subject: [PATCH] linting --- daft/filesystem.py | 3 +-- src/daft-io/src/lib.rs | 4 +++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/daft/filesystem.py b/daft/filesystem.py index 86849b66c5..b675c45ee4 100644 --- a/daft/filesystem.py +++ b/daft/filesystem.py @@ -2,7 +2,6 @@ import dataclasses import pathlib -import string import sys import urllib.parse from collections import defaultdict @@ -105,7 +104,7 @@ def get_protocol_from_path(path: str) -> str: parsed_scheme = parsed_scheme.lower() if parsed_scheme == "" or parsed_scheme is None: return "file" - if sys.platform == "win32" and len(parsed_scheme) == 1 and ('a' <= parsed_scheme) and (parsed_scheme <= 'z'): + if sys.platform == "win32" and len(parsed_scheme) == 1 and ("a" <= parsed_scheme) and (parsed_scheme <= "z"): return "file" return parsed_scheme diff --git a/src/daft-io/src/lib.rs b/src/daft-io/src/lib.rs index d630c97d11..3e07446216 100644 --- a/src/daft-io/src/lib.rs +++ b/src/daft-io/src/lib.rs @@ -245,7 +245,9 @@ fn parse_url(input: &str) -> Result<(SourceType, Cow<'_, str>)> { "az" | "abfs" => Ok((SourceType::AzureBlob, fixed_input)), "gcs" | "gs" => Ok((SourceType::GCS, fixed_input)), #[cfg(target_env = "msvc")] - _ if scheme.len() == 1 && ("a" <= scheme.as_str() && (scheme.as_str() <= "z")) => Ok((SourceType::File, Cow::Owned(format!("file://{input}")))), + _ if scheme.len() == 1 && ("a" <= scheme.as_str() && (scheme.as_str() <= "z")) => { + Ok((SourceType::File, Cow::Owned(format!("file://{input}")))) + } _ => Err(Error::NotImplementedSource { store: scheme }), } }