hysts HF staff commited on
Commit
6fd31fb
1 Parent(s): 7faab03
Files changed (1) hide show
  1. demo_list.py +18 -0
demo_list.py CHANGED
@@ -25,8 +25,23 @@ class DemoList:
25
  ['sdk', 'markdown'],
26
  ['sdk_version', 'str'],
27
  ['suggested_hardware', 'markdown'],
 
28
  ]
29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  def __init__(self):
31
  self.api = HfApi()
32
  self._raw_data = self.load_data()
@@ -65,6 +80,7 @@ class DemoList:
65
  info['likes'] = space_info.likes
66
  info['last_modified'] = space_info.lastModified
67
  info['status'] = space_info.runtime['stage']
 
68
  info['suggested_hardware'] = card.get('suggested_hardware', '')
69
  info['hardware'] = space_info.runtime['hardware']['current']
70
  if info['hardware'] is None:
@@ -129,6 +145,8 @@ class DemoList:
129
  self.to_div(row.sdk, 'sdk'),
130
  'sdk_version':
131
  row.sdk_version,
 
 
132
  }
133
  new_rows.append(new_row)
134
  df = pd.DataFrame(new_rows).loc[:, self.column_names]
 
25
  ['sdk', 'markdown'],
26
  ['sdk_version', 'str'],
27
  ['suggested_hardware', 'markdown'],
28
+ ['sleep_time', 'str'],
29
  ]
30
 
31
+ TO_TIME_STR = {
32
+ -1: '',
33
+ 300: '5 minutes',
34
+ 600: '10 minutes',
35
+ 900: '15 minutes',
36
+ 1800: '30 minutes',
37
+ 3600: '1 hour',
38
+ 36000: '10 hours',
39
+ 86400: '24 hours',
40
+ 172800: '48 hours',
41
+ 259200: '72 hours',
42
+ 604800: '1 week',
43
+ }
44
+
45
  def __init__(self):
46
  self.api = HfApi()
47
  self._raw_data = self.load_data()
 
80
  info['likes'] = space_info.likes
81
  info['last_modified'] = space_info.lastModified
82
  info['status'] = space_info.runtime['stage']
83
+ info['sleep_time'] = space_info.runtime['gcTimeout'] or -1
84
  info['suggested_hardware'] = card.get('suggested_hardware', '')
85
  info['hardware'] = space_info.runtime['hardware']['current']
86
  if info['hardware'] is None:
 
145
  self.to_div(row.sdk, 'sdk'),
146
  'sdk_version':
147
  row.sdk_version,
148
+ 'sleep_time':
149
+ self.TO_TIME_STR[row.sleep_time],
150
  }
151
  new_rows.append(new_row)
152
  df = pd.DataFrame(new_rows).loc[:, self.column_names]