From e9b36632c21ba937cb1c95b4929038b404cba729 Mon Sep 17 00:00:00 2001 From: Nancy Hung Date: Thu, 8 Feb 2024 21:57:10 -0800 Subject: [PATCH] fml --- composer/utils/object_store/uc_object_store.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/composer/utils/object_store/uc_object_store.py b/composer/utils/object_store/uc_object_store.py index 9b08530888..3ffe197fac 100644 --- a/composer/utils/object_store/uc_object_store.py +++ b/composer/utils/object_store/uc_object_store.py @@ -212,7 +212,7 @@ def get_object_size(self, object_name: str) -> int: # does not work. Once fixed, we will call the files API endpoint. We currently only use this # function in Composer and LLM-foundry to check the UC object's existence. self.client.api_client.do(method='HEAD', - path=f'{self._UC_VOLUME_FILES_API_ENDPOINT}/{self.prefix}/{object_name}', + path=f'{self._UC_VOLUME_FILES_API_ENDPOINT}/{self._get_object_path(object_name)}', headers={'Source': 'mosaicml/composer'}) return 1000000 # Dummy value, as we don't have a way to get the size of the file except DatabricksError as e: