Joshua Lochner commited on
Commit
4d4de75
1 Parent(s): 9f15397

Fix logging messages in predict script

Browse files
Files changed (1) hide show
  1. src/predict.py +2 -2
src/predict.py CHANGED
@@ -419,11 +419,11 @@ def main():
419
  predictions = predict(video_id, model, tokenizer,
420
  segmentation_args, classifier_args=classifier_args)
421
  except TranscriptError:
422
- logger.warning('No transcript available for', video_id, end='\n\n')
423
  continue
424
  video_url = f'https://www.youtube.com/watch?v={video_id}'
425
  if not predictions:
426
- logger.info('No predictions found for', video_url, end='\n\n')
427
  continue
428
 
429
  # TODO use predict_args.output_as_json
 
419
  predictions = predict(video_id, model, tokenizer,
420
  segmentation_args, classifier_args=classifier_args)
421
  except TranscriptError:
422
+ logger.warning(f'No transcript available for {video_id}')
423
  continue
424
  video_url = f'https://www.youtube.com/watch?v={video_id}'
425
  if not predictions:
426
+ logger.info(f'No predictions found for {video_url}')
427
  continue
428
 
429
  # TODO use predict_args.output_as_json