Add URL substitution to registered uploaded files when reporting an e…

…vent
allegroai · May 4, 2024 · 81c02db · 81c02db
1 parent f267466
commit 81c02db
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/clearml/backend_interface/metrics/interface.py b/clearml/backend_interface/metrics/interface.py
@@ -136,6 +136,15 @@ def update_and_get_file_entry(ev):
             kwargs = {}
             if entry:
                 key, url = ev.get_target_full_upload_uri(storage_uri, self.storage_key_prefix, quote_uri=False)
+
+                orig_url = url
+                try:
+                    storage = self._get_storage(url)
+                    if storage:
+                        url = storage._apply_url_substitutions(url)
+                except Exception as err:
+                    self._get_logger().warning("Failed applying URL substitutions on {} ({})".format(orig_url, err))
+
                 kwargs[entry.key_prop] = key
                 kwargs[entry.url_prop] = url
                 if not entry.stream: