def collect_references(config: V1Operation, path_context: str = None):
if config.has_component_reference:
return config
elif config.has_hub_reference:
component = ConfigSpec.get_from(config.hub_ref, "hub").read()
elif config.has_url_reference:
component = ConfigSpec.get_from(config.url_ref, "url").read()
elif config.has_path_reference:
path_ref = config.path_ref
if path_context:
path_ref = os.path.join(
os.path.dirname(os.path.abspath(path_context)), path_ref)
component = ConfigSpec.get_from(path_ref).read()
else:
raise PolyaxonfileError("Operation found without component")
component = get_specification(data=component)
if component.kind != kinds.COMPONENT:
if config.has_url_reference:
ref_type = "Url ref"
ref = config.url_ref
else:
ref_type = "Path ref"
ref = config.path_ref
raise PolyaxonfileError(
"the reference ({}) `{}` is of kind `{}`, it should be a `{}`".
format(ref, ref_type, component.kind, kinds.COMPONENT))
config.component = component
if component.is_dag_run:
component.run.collect_components()
return config
def collect_references(config: V1Operation):
if config.has_component_reference or config.has_hub_reference:
return config
elif config.has_url_reference:
component = ConfigSpec.get_from(config.url_ref, "url").read()
elif config.has_path_reference:
component = ConfigSpec.get_from(config.path_ref).read()
else:
raise PolyaxonfileError("Operation found without component")
component = get_specification(data=component)
config.component = component
return config
