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,8 +34,11 @@ def filter_pipelines(usage_dict, usage_cutoff=10000):
if usage < usage_cutoff:
continue
if "Pipeline" in diffusers_object:
output.append(diffusers_object)
is_diffusers_pipeline = hasattr(diffusers.pipelines, diffusers_object)
if not is_diffusers_pipeline:
continue
output.append(diffusers_object)
return output
......@@ -71,6 +74,7 @@ def fetch_pipeline_modules_to_test():
test_modules = []
for pipeline_name in pipeline_objects:
module = getattr(diffusers, pipeline_name)
test_module = module.__module__.split(".")[-2].strip()
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