From be3ff851c687695abe39cdcfc679fbd4015ecb30 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 27 Oct 2023 18:48:07 +0530 Subject: [PATCH] fix: test --- tests/others/test_workflows.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/others/test_workflows.py b/tests/others/test_workflows.py index 174182bd88..7cad91786c 100644 --- a/tests/others/test_workflows.py +++ b/tests/others/test_workflows.py @@ -22,7 +22,7 @@ import uuid import numpy as np import torch from huggingface_hub import delete_repo, hf_hub_download -from test_utils import TOKEN, is_staging_test +from test_utils import TOKEN, USER, is_staging_test from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer from diffusers import ( @@ -162,7 +162,7 @@ class WorkflowPushToHubTester(unittest.TestCase): workflow = populate_workflow_from_pipeline(list(inputs.keys()), inputs, None) workflow.push_to_hub(self.repo_id, token=TOKEN) - local_path = hf_hub_download(repo_id=self.repo_id, filename=WORKFLOW_NAME, token=TOKEN) + local_path = hf_hub_download(repo_id=f"{USER}/{self.repo_id}", filename=WORKFLOW_NAME, token=TOKEN) with open(local_path) as f: locally_loaded_workflow = json.load(f)