Joshua Lochner commited on
Commit
c4f250e
·
1 Parent(s): aa018be

Initialize logging in each script

Browse files
Files changed (5) hide show
  1. src/evaluate.py +1 -0
  2. src/predict.py +1 -0
  3. src/preprocess.py +1 -0
  4. src/train.py +1 -1
  5. src/utils.py +1 -0
src/evaluate.py CHANGED
@@ -17,6 +17,7 @@ from shared import seconds_to_time
17
  from urllib.parse import quote
18
  import logging
19
 
 
20
  logger = logging.getLogger(__name__)
21
 
22
 
 
17
  from urllib.parse import quote
18
  import logging
19
 
20
+ logging.basicConfig()
21
  logger = logging.getLogger(__name__)
22
 
23
 
src/predict.py CHANGED
@@ -25,6 +25,7 @@ import preprocess
25
  from errors import PredictionException, TranscriptError, ModelLoadError, ClassifierLoadError
26
  from model import ModelArguments, get_classifier_vectorizer, get_model_tokenizer
27
 
 
28
  logger = logging.getLogger(__name__)
29
 
30
  # Public innertube key (b64 encoded so that it is not incorrectly flagged)
 
25
  from errors import PredictionException, TranscriptError, ModelLoadError, ClassifierLoadError
26
  from model import ModelArguments, get_classifier_vectorizer, get_model_tokenizer
27
 
28
+ logging.basicConfig()
29
  logger = logging.getLogger(__name__)
30
 
31
  # Public innertube key (b64 encoded so that it is not incorrectly flagged)
src/preprocess.py CHANGED
@@ -20,6 +20,7 @@ import time
20
  import requests
21
 
22
 
 
23
  logger = logging.getLogger(__name__)
24
 
25
 
 
20
  import requests
21
 
22
 
23
+ logging.basicConfig()
24
  logger = logging.getLogger(__name__)
25
 
26
 
src/train.py CHANGED
@@ -32,7 +32,7 @@ require_version('datasets>=1.8.0',
32
 
33
  os.environ['WANDB_DISABLED'] = 'true'
34
 
35
-
36
  logger = logging.getLogger(__name__)
37
 
38
  # Setup logging
 
32
 
33
  os.environ['WANDB_DISABLED'] = 'true'
34
 
35
+ logging.basicConfig()
36
  logger = logging.getLogger(__name__)
37
 
38
  # Setup logging
src/utils.py CHANGED
@@ -1,6 +1,7 @@
1
  import re
2
  import logging
3
 
 
4
  logger = logging.getLogger(__name__)
5
 
6
 
 
1
  import re
2
  import logging
3
 
4
+ logging.basicConfig()
5
  logger = logging.getLogger(__name__)
6
 
7