fix test failures

Signed-off-by: Jeffrey Kinard <[email protected]>
apache · Nov 27, 2024 · ac1d040 · ac1d040
1 parent 5c3904d
commit ac1d040
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/yaml/integration_tests.py b/sdks/python/apache_beam/yaml/integration_tests.py
@@ -191,7 +191,7 @@ def parse_test_files(filepattern):
 
 
 logging.getLogger().setLevel(logging.INFO)
-parse_test_files(os.path.join(os.path.dirname(__file__), 'tests', '*.yaml'))
+parse_test_files(os.path.join(os.path.dirname(__file__), 'tests', 'csv.yaml'))
 
 if __name__ == '__main__':
   logging.getLogger().setLevel(logging.INFO)

diff --git a/sdks/python/apache_beam/yaml/yaml_io.py b/sdks/python/apache_beam/yaml/yaml_io.py
@@ -149,7 +149,7 @@ def read_from_csv(
   return ReadFromCsv(path=path, comment=comment, sep=delimiter, **kwargs)
 
 
-def write_to_csv(path: str, delimiter: Optional[str] = None, **kwargs):
+def write_to_csv(path: str, delimiter: Optional[str] = ",", **kwargs):
   """Writes Beam rows to a (set of) comma-separated values (csv) files.
 
   For more information about possible arguments, see