qgallouedec HF staff commited on
Commit
5028d3c
1 Parent(s): a2f004f

fix generator

Browse files
Files changed (1) hide show
  1. src/backend.py +1 -1
src/backend.py CHANGED
@@ -133,7 +133,7 @@ def pattern_match(patterns, source_list):
133
 
134
  def _backend_routine():
135
  # List only the text classification models
136
- rl_models = API.list_models(filter=["reinforcement-learning"])
137
  logger.info(f"Found {len(rl_models)} RL models")
138
 
139
  compatible_models = []
 
133
 
134
  def _backend_routine():
135
  # List only the text classification models
136
+ rl_models = list(API.list_models(filter=["reinforcement-learning"]))
137
  logger.info(f"Found {len(rl_models)} RL models")
138
 
139
  compatible_models = []