From 38bebdca6b3aa142d9bb02deafb5df1bcc1a6b81 Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@meta.com>
Date: Fri, 22 Mar 2024 12:55:34 -0400
Subject: [PATCH] Simple fix on extended configs

---
 torchbenchmark/util/framework/timm/extended_configs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchbenchmark/util/framework/timm/extended_configs.py b/torchbenchmark/util/framework/timm/extended_configs.py
index dbf67fce0f..6315404b83 100644
--- a/torchbenchmark/util/framework/timm/extended_configs.py
+++ b/torchbenchmark/util/framework/timm/extended_configs.py
@@ -19,7 +19,7 @@ def is_extended_timm_models(model_name: str) -> bool:
     return model_name in TIMM_MODELS
 
 def list_extended_timm_models() -> List[str]:
-    return TIMM_MODELS.keys()
+    return list(TIMM_MODELS.keys())
 
 # TODO - Figure out the reason of cold start memory spike
 BATCH_SIZE_DIVISORS = {