Apply Literal to to_pandas()

deephaven · Nov 10, 2023 · 80340a4 · 80340a4
1 parent 2817070
commit 80340a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/py/server/deephaven/pandas.py b/py/server/deephaven/pandas.py
@@ -3,7 +3,7 @@
 #
 
 """ This module supports the conversion between Deephaven tables and pandas DataFrames. """
-from typing import List, Dict, Tuple
+from typing import List, Dict, Tuple, Literal
 
 import jpy
 import numpy as np
@@ -112,8 +112,8 @@ def _column_to_series(table: Table, col_def: Column, conv_null: bool) -> pd.Seri
 }
 
 
-def to_pandas(table: Table, cols: List[str] = None, dtype_backend: str = None, conv_null: bool = True) -> \
-        pd.DataFrame:
+def to_pandas(table: Table, cols: List[str] = None, dtype_backend: Literal[None, "pyarrow", "numpy_nullable"] = None,
+              conv_null: bool = True) -> pd.DataFrame:
     """Produces a pandas DataFrame from a table.
 
     Note that the **entire table** is going to be cloned into memory, so the total number of entries in the table

diff --git a/py/server/deephaven/table_factory.py b/py/server/deephaven/table_factory.py
@@ -36,7 +36,6 @@
 _JFunctionGeneratedTableFactory = jpy.get_type("io.deephaven.engine.table.impl.util.FunctionGeneratedTableFactory")
 
 
-
 def empty_table(size: int) -> Table:
     """Creates a table with rows but no columns.
 
@@ -395,7 +394,6 @@ def function_generated_table(table_generator: Callable[..., Table],
         if exec_ctx is None:
             raise ValueError("No execution context is available and exec_ctx was not provided! ")
 
-
     def table_generator_function():
         with exec_ctx:
             result = table_generator(*args, **kwargs)