zhuwq0 commited on
Commit
fbe7f57
·
1 Parent(s): d944358
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -61,7 +61,7 @@ class PreTrainedPipeline():
61
  # picks = [{k: v for k, v in pick.items() if k in ["station_id", "phase_time", "phase_score", "phase_type", "dt"]} for pick in picks]
62
 
63
  # return picks
64
- return [[{"label": "debug", "score": 0.1}]]
65
 
66
 
67
  if __name__ == "__main__":
 
61
  # picks = [{k: v for k, v in pick.items() if k in ["station_id", "phase_time", "phase_score", "phase_type", "dt"]} for pick in picks]
62
 
63
  # return picks
64
+ return [[picks, {"label": "debug", "score": 0.1}]]
65
 
66
 
67
  if __name__ == "__main__":