Unverified Commit 28e8d1f6 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix pipe fetcher for slow tests (#5424)

* fix pipe fetcher

* filter out community pipelines
parent 98a0712d
...@@ -34,7 +34,10 @@ def filter_pipelines(usage_dict, usage_cutoff=10000): ...@@ -34,7 +34,10 @@ def filter_pipelines(usage_dict, usage_cutoff=10000):
if usage < usage_cutoff: if usage < usage_cutoff:
continue continue
if "Pipeline" in diffusers_object: is_diffusers_pipeline = hasattr(diffusers.pipelines, diffusers_object)
if not is_diffusers_pipeline:
continue
output.append(diffusers_object) output.append(diffusers_object)
return output return output
...@@ -71,6 +74,7 @@ def fetch_pipeline_modules_to_test(): ...@@ -71,6 +74,7 @@ def fetch_pipeline_modules_to_test():
test_modules = [] test_modules = []
for pipeline_name in pipeline_objects: for pipeline_name in pipeline_objects:
module = getattr(diffusers, pipeline_name) module = getattr(diffusers, pipeline_name)
test_module = module.__module__.split(".")[-2].strip() test_module = module.__module__.split(".")[-2].strip()
test_modules.append(test_module) test_modules.append(test_module)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment