Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
qq_36265860
spleeter
提交
6c9a9064
S
spleeter
项目概览
qq_36265860
/
spleeter
与 Fork 源项目一致
从无法访问的项目Fork
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
S
spleeter
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
6c9a9064
编写于
12月 08, 2020
作者:
F
Faylixe
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
✨
add dedicated Typer logger
上级
8a0e4200
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
1 deletion
+10
-1
spleeter/utils/logging.py
spleeter/utils/logging.py
+9
-1
tests/test_train.py
tests/test_train.py
+1
-0
未找到文件。
spleeter/utils/logging.py
浏览文件 @
6c9a9064
...
...
@@ -11,6 +11,7 @@ from os import environ
# pyright: reportMissingImports=false
# pylint: disable=import-error
from
tensorflow.compat.v1
import
logging
as
tflogging
from
typer
import
echo
# pylint: enable=import-error
__email__
=
'spleeter@deezer.com'
...
...
@@ -18,8 +19,15 @@ __author__ = 'Deezer Research'
__license__
=
'MIT License'
class
TyperLoggerHandler
(
logging
.
Handler
):
""" A custom logger handler that use Typer echo. """
def
emit
(
self
,
record
:
logging
.
LogRecord
)
->
None
:
echo
(
self
.
format
(
record
))
formatter
=
logging
.
Formatter
(
'%(levelname)s:%(name)s:%(message)s'
)
handler
=
logging
.
Stream
Handler
()
handler
=
TyperLogger
Handler
()
handler
.
setFormatter
(
formatter
)
logger
:
logging
.
Logger
=
logging
.
getLogger
(
'spleeter'
)
logger
.
addHandler
(
handler
)
...
...
tests/test_train.py
浏览文件 @
6c9a9064
...
...
@@ -101,6 +101,7 @@ def test_train():
'-p'
,
'useless_config.json'
,
'-d'
,
path
])
raise
IOError
(
f
'STDOUT:
{
result
.
stdout
}
, STDERR:
{
result
.
stderr
}
'
)
assert
result
.
exit_code
==
0
# assert that model checkpoint was created.
assert
os
.
path
.
exists
(
join
(
path
,
'model'
,
'model.ckpt-10.index'
))
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录