lixuejing commited on
Commit
4772f92
·
1 Parent(s): a28724a
Files changed (1) hide show
  1. src/tools/collections.py +5 -3
src/tools/collections.py CHANGED
@@ -2,7 +2,7 @@ import os
2
 
3
  import pandas as pd
4
  from huggingface_hub import add_collection_item, delete_collection_item, get_collection, update_collection_item
5
- from huggingface_hub.utils._errors import HfHubHTTPError
6
  from pandas import DataFrame
7
 
8
  from src.display.utils import AutoEvalColumn, ModelType
@@ -69,7 +69,8 @@ def update_collections(df: DataFrame):
69
  cur_len_collection = len(collection.items)
70
  cur_best_models.append(model)
71
  break
72
- except HfHubHTTPError:
 
73
  continue
74
 
75
  collection = get_collection(PATH_TO_COLLECTION, token=TOKEN)
@@ -79,5 +80,6 @@ def update_collections(df: DataFrame):
79
  delete_collection_item(
80
  collection_slug=PATH_TO_COLLECTION, item_object_id=item.item_object_id, token=TOKEN
81
  )
82
- except HfHubHTTPError:
 
83
  continue
 
2
 
3
  import pandas as pd
4
  from huggingface_hub import add_collection_item, delete_collection_item, get_collection, update_collection_item
5
+ #from huggingface_hub.utils._errors import HfHubHTTPError
6
  from pandas import DataFrame
7
 
8
  from src.display.utils import AutoEvalColumn, ModelType
 
69
  cur_len_collection = len(collection.items)
70
  cur_best_models.append(model)
71
  break
72
+ #except HfHubHTTPError:
73
+ except:
74
  continue
75
 
76
  collection = get_collection(PATH_TO_COLLECTION, token=TOKEN)
 
80
  delete_collection_item(
81
  collection_slug=PATH_TO_COLLECTION, item_object_id=item.item_object_id, token=TOKEN
82
  )
83
+ #except HfHubHTTPError:
84
+ except:
85
  continue