andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Add colours to logging

Browse source 
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-04-10 09:41:04 +02:00
commit 3730a884b0
1 changed files with 84 additions and 268 deletions
Download patch file Download diff file Expand all files Collapse all files

6
good_ai/src/good_ai/good_ai/models/save_model.py
View file

@ -1,13 +1,11 @@
import logging
from pathlib import Path
from typing import Optional, Union
from joblib import dump
from good_ai.good_ai.context.get_context import get_context
from good_ai.open_s3 import LargeFile
logger = logging.getLogger("models")
from ..context import get_context
def save_model(
@ -23,6 +21,6 @@ def save_model(
with file as f:
dump(model, f)
logger.info(f"Model {key} uploaded with version {file.version}")
get_context().logger.info(f"Model {key} uploaded with version {file.version}")
return file.version