From b61bf7148d4411bb81ddcf6350a3ef1427213649 Mon Sep 17 00:00:00 2001 From: shuaiqing Date: Sun, 14 Jan 2024 16:54:10 +0800 Subject: [PATCH] update config --- easymocap/config/yacs.py | 5 ++++- setup.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/easymocap/config/yacs.py b/easymocap/config/yacs.py index b7d4542..9219f63 100644 --- a/easymocap/config/yacs.py +++ b/easymocap/config/yacs.py @@ -171,7 +171,10 @@ class CfgNode(dict): return "{}" for k, v in self.items(): seperator = "\n" if isinstance(v, CfgNode) and len(v.keys()) > 0 else " " - attr_str = "{}:{}{}".format(str(k), seperator, str(v)) + if isinstance(v, float): + attr_str = "{}:{}{:f}".format(str(k), seperator, v) + else: + attr_str = "{}:{}{}".format(str(k), seperator, str(v)) attr_str = _indent(attr_str, 4) s.append(attr_str) r += "\n".join(s) diff --git a/setup.py b/setup.py index 444d709..8938248 100644 --- a/setup.py +++ b/setup.py @@ -16,6 +16,7 @@ setup( # test_suite='setup.test_all', packages=[ 'easymocap', + 'easymocap.config', 'easymocap.dataset', 'easymocap.smplmodel', 'easymocap.pyfitting',