diff options
| author | iaz1607 <iaz1607@yandex-team.com> | 2023-11-30 12:16:39 +0300 |
|---|---|---|
| committer | iaz1607 <iaz1607@yandex-team.com> | 2023-11-30 12:56:46 +0300 |
| commit | 8951ddf780e02616cdb2ec54a02bc354e8507c0f (patch) | |
| tree | 478097488957d3b554e25868c972a959bb40d78e /build/scripts/build_catboost.py | |
| parent | a5acb7aa4ca5a4603215e878eb0cad786793262b (diff) | |
| download | ydb-8951ddf780e02616cdb2ec54a02bc354e8507c0f.tar.gz | |
`build/scripts` ya style --py
Diffstat (limited to 'build/scripts/build_catboost.py')
| -rwxr-xr-x | build/scripts/build_catboost.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/build/scripts/build_catboost.py b/build/scripts/build_catboost.py index 78334fc5f7..81d4e795a0 100755 --- a/build/scripts/build_catboost.py +++ b/build/scripts/build_catboost.py @@ -1,9 +1,9 @@ import sys import os import shutil -import re import subprocess + def get_value(val): dct = val.split('=', 1) if len(dct) > 1: @@ -13,7 +13,6 @@ def get_value(val): class BuildCbBase(object): def run(self, cbmodel, cbname, cb_cpp_path): - data_prefix = "CB_External_" data = data_prefix + cbname datasize = data + "Size" @@ -33,17 +32,20 @@ class BuildCbBase(object): cb_cpp_tmp.write(" extern const ui32 {1}{0}Size;\n".format(cbname, data_prefix)) cb_cpp_tmp.write(" }\n") cb_cpp_tmp.write("}\n") - archiverCall = subprocess.Popen([self.archiver, "-q", "-p", "-o", ro_data_path, cbmodel], stdout=None, stderr=subprocess.PIPE) + archiverCall = subprocess.Popen( + [self.archiver, "-q", "-p", "-o", ro_data_path, cbmodel], stdout=None, stderr=subprocess.PIPE + ) archiverCall.wait() cb_cpp_tmp.write("extern {0} {1};\n".format(cbtype, cbname)) cb_cpp_tmp.write("{0} {1}{2};".format(cbtype, cbname, cbload)) cb_cpp_tmp.close() shutil.move(cb_cpp_tmp_path, cb_cpp_path) + class BuildCb(BuildCbBase): def run(self, argv): if len(argv) < 5: - print >>sys.stderr, "BuildCb.Run(<ARCADIA_ROOT> <archiver> <mninfo> <mnname> <cppOutput> [params...])" + print >> sys.stderr, "BuildCb.Run(<ARCADIA_ROOT> <archiver> <mninfo> <mnname> <cppOutput> [params...])" sys.exit(1) self.SrcRoot = argv[0] @@ -62,10 +64,10 @@ def build_cb_f(argv): if __name__ == '__main__': if len(sys.argv) < 2: - print >>sys.stderr, "Usage: build_cb.py <funcName> <args...>" + print >> sys.stderr, "Usage: build_cb.py <funcName> <args...>" sys.exit(1) - if (sys.argv[2:]): + if sys.argv[2:]: globals()[sys.argv[1]](sys.argv[2:]) else: globals()[sys.argv[1]]() |