fix test failures

Signed-off-by: Jeffrey Kinard <[email protected]>
apache · Dec 4, 2024 · a024647 · a024647
1 parent bf70181
commit a024647
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