Fixes Pandas CSV Reader Writer names

CSV was not capitalized. Fixing that.

hamilton/plugins/pandas_extensions.py

@@ -83,7 +83,7 @@ def save_data(self, data: DATAFRAME_TYPE) -> Dict[str, Any]:
 
 
 @dataclasses.dataclass
-class PandasCsvReader(DataLoader):
+class PandasCSVReader(DataLoader):
     """
     Class that handles saving CSV files with pandas.
     Maps to https://pandas.pydata.org/docs/reference/api/pandas.read_csv.html
@@ -251,8 +251,8 @@ def name(cls) -> str:
 
 
 @dataclasses.dataclass
-class PandasCsvWriter(DataSaver):
-    """Class that handles saving Csv files with pandas.
+class PandasCSVWriter(DataSaver):
+    """Class that handles saving CSV files with pandas.
     Maps to https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.to_csv.html
     """
 
@@ -1323,8 +1323,8 @@ def name(cls) -> str:
 def register_data_loaders():
     """Function to register the data loaders for this extension."""
     for loader in [
-        PandasCsvReader,
-        PandasCsvWriter,
+        PandasCSVReader,
+        PandasCSVWriter,
         PandasParquetReader,
         PandasParquetWriter,
         PandasPickleReader,

tests/plugins/test_pandas_extensions.py

@@ -9,8 +9,8 @@
 from sqlalchemy import create_engine
 
 from hamilton.plugins.pandas_extensions import (
-    PandasCsvReader,
-    PandasCsvWriter,
+    PandasCSVReader,
+    PandasCSVWriter,
     PandasFeatherReader,
     PandasFeatherWriter,
     PandasHtmlReader,
@@ -188,19 +188,19 @@ def test_pandas_feather_writer(tmp_path: pathlib.Path) -> None:
 
 def test_pandas_csv_reader(tmp_path: pathlib.Path) -> None:
     path_to_test = "tests/resources/data/test_load_from_data.csv"
-    reader = PandasCsvReader(path=path_to_test)
+    reader = PandasCSVReader(path=path_to_test)
     df, metadata = reader.load_data(pd.DataFrame)
 
-    assert PandasCsvReader.applicable_types() == [pd.DataFrame]
+    assert PandasCSVReader.applicable_types() == [pd.DataFrame]
     assert df.loc[0, "firstName"] == "John"
     assert df.shape == (3, 5)
 
 
 def test_pandas_csv_writer(tmp_path: pathlib.Path) -> None:
     file_path = tmp_path / "test.csv"
-    writer = PandasCsvWriter(path=file_path)
+    writer = PandasCSVWriter(path=file_path)
     metadata = writer.save_data(pd.DataFrame(data={"col1": [1, 2], "col2": [4, 3]}))
 
-    assert PandasCsvWriter.applicable_types() == [pd.DataFrame]
+    assert PandasCSVWriter.applicable_types() == [pd.DataFrame]
     assert file_path.exists()
     assert metadata["path"] == file_path