move get_random_row to new file

placeTW · Jul 14, 2024 · b86027f · b86027f
1 parent 7a0c883
commit b86027f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/commands/taiwanese/read_embree_csv.py b/commands/taiwanese/read_embree_csv.py
@@ -34,7 +34,4 @@ def read_embree_csv_raw(
     df = pd.read_csv(filepath, encoding='utf-8', index_col=index_col)
     if cols_to_keep:
         df = df[list(cols_to_keep)]
-    return df
-
-def get_random_row(df: pd.DataFrame) -> pd.Series:
-    return df.sample().iloc[0]
+    return df
diff --git a/commands/taiwanese/taiwanese_entry.py b/commands/taiwanese/taiwanese_entry.py
@@ -0,0 +1,5 @@
+import pandas as pd
+from pathlib import Path
+
+def get_random_row(df: pd.DataFrame) -> pd.Series:
+    return df.sample().iloc[0]