patch gcs path parsing

jitkasempin · Apr 26, 2019 · fecf71a · fecf71a
1 parent 82f2426
commit fecf71a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...les/dataflow-data-generator/data-generator-pipeline/data_generator/PrettyDataGenerator.py b/...les/dataflow-data-generator/data-generator-pipeline/data_generator/PrettyDataGenerator.py
@@ -93,7 +93,7 @@ def __init__(self, bq_schema_filename=None, input_bq_table=None,
             try:
                 # Handles json from google cloud storage or local.
                 if bq_schema_filename.find('gs://') == 0:
-                    bkt, path = bq_schema_filename.strip('gs://').split('/', 1)
+                    bkt, path = bq_schema_filename.replace('gs://', '').split('/', 1)
                     client = gcs.Client()
                     bucket = client.get_bucket(bkt)