diff --git a/sdk/python/kfp/components/_components.py b/sdk/python/kfp/components/_components.py
index fd72032f3a8..7e482d52cb9 100644
--- a/sdk/python/kfp/components/_components.py
+++ b/sdk/python/kfp/components/_components.py
@@ -270,5 +270,5 @@ def component_default_to_func_default(component_default: str, is_optional: bool)
         factory_function_parameters,
         documentation='\n'.join(func_docstring_lines),
         func_name=name,
-        func_filename=component_filename
+        func_filename=component_filename if (component_filename and component_filename.endswith('.yaml')) else None,
     )