davidberenstein1957 HF staff commited on
Commit
5ade45a
1 Parent(s): 4883d00

fix: add explicit repo id

Browse files
src/distilabel_dataset_generator/apps/sft.py CHANGED
@@ -176,15 +176,17 @@ def upload_pipeline_code(
176
  oauth_token: Union[OAuthToken, None] = None,
177
  progress=gr.Progress(),
178
  ):
 
179
  progress(0.1, desc="Uploading pipeline code")
180
  with io.BytesIO(pipeline_code.encode("utf-8")) as f:
181
  upload_file(
182
  path_or_fileobj=f,
183
  path_in_repo="pipeline.py",
184
- repo_id=f"{org_name}/{repo_name}",
185
  repo_type="dataset",
186
  token=oauth_token.token,
187
  commit_message="Include pipeline script",
 
188
  )
189
  progress(1.0, desc="Pipeline code uploaded")
190
 
 
176
  oauth_token: Union[OAuthToken, None] = None,
177
  progress=gr.Progress(),
178
  ):
179
+ repo_id = _check_push_to_hub(org_name, repo_name)
180
  progress(0.1, desc="Uploading pipeline code")
181
  with io.BytesIO(pipeline_code.encode("utf-8")) as f:
182
  upload_file(
183
  path_or_fileobj=f,
184
  path_in_repo="pipeline.py",
185
+ repo_id=repo_id,
186
  repo_type="dataset",
187
  token=oauth_token.token,
188
  commit_message="Include pipeline script",
189
+ create_pr=False,
190
  )
191
  progress(1.0, desc="Pipeline code uploaded")
192