ran ruff

Author: jsai28

python/datafusion/context.py

@@ -499,7 +499,7 @@ def __init__(
         self.ctx = SessionContextInternal(config, runtime)
 
     @classmethod
-    def global_ctx(cls) -> "SessionContext":
+    def global_ctx(cls) -> SessionContext:
         """Retrieve the global context as a `SessionContext` wrapper.
 
         Returns:
@@ -510,7 +510,7 @@ def global_ctx(cls) -> "SessionContext":
         wrapper.ctx = internal_ctx
         return wrapper
 
-    def enable_url_table(self) -> "SessionContext":
+    def enable_url_table(self) -> SessionContext:
         """Control if local files can be queried as tables.
 
         Returns:

python/datafusion/io.py

@@ -23,11 +23,10 @@
 
 import pyarrow
 
+from datafusion.context import SessionContext
 from datafusion.dataframe import DataFrame
 from datafusion.expr import Expr
 
-from datafusion.context import SessionContext
-
 
 def read_parquet(
     path: str | pathlib.Path,