Uploaded image for project: 'Beam'
  1. Beam
  2. BEAM-12701

Converting two deferred dataframes to csv in the same pipeline causes a PCollection label collision

Details

    • Bug
    • Status: Triage Needed
    • P2
    • Resolution: Fixed
    • 2.31.0, 2.32.0, 2.33.0
    • 2.34.0
    • dsl-dataframe
    • None

    Description

       

      If you use  the to_csv of the DeferredDataFrame twice in a single pipeline like this : 

      df1 = pd.DataFrame.from_records({"a":"b"}, index=[0])
      df2 = pd.DataFrame.from_records({"a":"b"}, index=[0])
      
      with beam.Pipeline() as p:
       df1 = to_dataframe(to_pcollection(df1, pipeline=p), label="df1")
       df2 = to_dataframe(to_pcollection(df2, pipeline=p), label="df2")
      
       df1.to_csv("test.csv")
       df2.to_csv("test2.csv")

      You get this error on the second to_csv call

       

      RuntimeError: A transform with label "ToPCollection(df)" already exists in the pipeline. To apply a transform with a specified label write pvalue | "label" >> transform
      
      

      I think it comes from the fact that to_csv  is calling a  to_pcollection without any label, causing to infer an identical label for both to_csv function calls. 

       

       

      Attachments

        Issue Links

          Activity

            People

              edusan Eduardo Sánchez López
              BigJerBD Jérémie Bigras-Dunberry
              Votes:
              1 Vote for this issue
              Watchers:
              1 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved:

                Time Tracking

                  Estimated:
                  Original Estimate - Not Specified
                  Not Specified
                  Remaining:
                  Remaining Estimate - 0h
                  0h
                  Logged:
                  Time Spent - 7.5h
                  7.5h