simplify code

activitysim/core/configuration/top.py

@@ -3,9 +3,10 @@
 from pathlib import Path
 from typing import Any, Literal
 
-from activitysim.core.configuration.base import PydanticBase, Union
 from pydantic import validator
 
+from activitysim.core.configuration.base import PydanticBase, Union
+
 
 class InputTable(PydanticBase):
     """
@@ -120,9 +121,9 @@ class OutputTables(PydanticBase):
     h5_store: bool = False
     """Write tables into a single HDF5 store instead of individual CSVs."""
 
-    file_type: str = "csv"
+    file_type: Literal["csv", "parquet", "h5"] = "csv"
     """
-    Specifies the file type for output tables. Options are limited to 'csv', 
+    Specifies the file type for output tables. Options are limited to 'csv',
     'h5' or 'parquet'. Only applied if h5_store is set to False."""
 
     action: str
@@ -150,15 +151,6 @@ class OutputTables(PydanticBase):
     applied to any output tables.
     """
 
-    @validator("file_type")
-    def method_is_valid(cls, method: str) -> str:
-        """Validates file_type setting."""
-
-        allowed_set = {"csv", "h5", "parquet"}
-        if method not in allowed_set:
-            raise ValueError(f"must be in {allowed_set}, got '{method}'")
-        return method
-
 
 class MultiprocessStepSlice(PydanticBase, extra="forbid"):
     """

activitysim/core/steps/output.py

@@ -413,5 +413,7 @@ def map_func(x):
             # include the index if it has a name or is a MultiIndex
             if file_type == "csv":
                 csv.write_csv(dt, file_path)
-            else:
+            elif file_type == "parquet":
                 parquet.write_table(dt, file_path)
+            else:
+                raise ValueError(f"unknown file_type {file_type}")