From fc0d5648c26bd2a827770ee41d0639362091869e Mon Sep 17 00:00:00 2001 From: Albert Torosyan Date: Wed, 27 Sep 2023 18:30:26 +0400 Subject: [PATCH] [fix] Use correct value of container.type in query context --- src/python/aim/_sdk/query_utils.py | 3 ++- src/python/aim/_sdk/repo.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/python/aim/_sdk/query_utils.py b/src/python/aim/_sdk/query_utils.py index 3d3e9c51bd..552db19a05 100644 --- a/src/python/aim/_sdk/query_utils.py +++ b/src/python/aim/_sdk/query_utils.py @@ -5,6 +5,7 @@ from aim._core.storage.proxy import AimObjectProxy from aim._core.storage.types import SafeNone +from aim._sdk.constants import KeyNames from typing import TYPE_CHECKING if TYPE_CHECKING: @@ -44,7 +45,7 @@ def hash(self): @property def type(self): - return self._cont_tree['info_', 'cont_type'] + return self._cont_tree[KeyNames.INFO_PREFIX, KeyNames.CONTAINER_TYPE].split('->')[-1] def __getattr__(self, item): return self[item] # fallback to __getitem__ diff --git a/src/python/aim/_sdk/repo.py b/src/python/aim/_sdk/repo.py index 3671ee5908..be3d8a0142 100644 --- a/src/python/aim/_sdk/repo.py +++ b/src/python/aim/_sdk/repo.py @@ -404,7 +404,7 @@ def copy_trees(): for key, val in source_meta_container_attrs_tree_collected.items(): dest_meta_attrs_tree.merge(key, val) - cont_type = source_meta_container_tree_collected[KeyNames.INFO_PREFIX]['cont_type'].split('->')[-1] + cont_type = source_meta_container_tree_collected[KeyNames.INFO_PREFIX][KeyNames.CONTAINER_TYPE].split('->')[-1] dest_meta_tree[('cont_types_map', container_hash)] = source_meta_tree[('cont_types_map', container_hash)] dest_meta_tree[('containers', cont_type)] = 1