diff --git a/katrain/core/engine.py b/katrain/core/engine.py index 6ffb786..875a9ff 100644 --- a/katrain/core/engine.py +++ b/katrain/core/engine.py @@ -37,7 +37,7 @@ class KataGoEngine: self.query_counter = 0 self.katago_process = None self.base_priority = 0 - self.override_settings = {'reportAnalysisWinratesAs':'BLACK'} # force these settings + self.override_settings = {"reportAnalysisWinratesAs": "BLACK"} # force these settings self._lock = threading.Lock() self.analysis_thread = None self.stderr_thread = None diff --git a/katrain/gui/popups.py b/katrain/gui/popups.py index e481ffb..c8af325 100644 --- a/katrain/gui/popups.py +++ b/katrain/gui/popups.py @@ -307,12 +307,12 @@ class AIPopup(QuickConfigGui): for _ in range((self.max_options - len(mode_settings)) * 2): self.options_grid.add_widget(Label()) + class ConfigPopup(QuickConfigGui): def __init__(self, katrain): super().__init__(katrain) self.paths = [self.katrain.config("engine/model"), "katrain/models", "~/.katrain"] - def build_and_set_properties(self, *_args): super().build_and_set_properties() # self.check_models() @@ -336,7 +336,7 @@ class ConfigPopup(QuickConfigGui): for ftype in ["*.bin.gz", "*.txt.gz"] for f in glob.glob(slashpath + "/" + ftype) ] - print(path,files) + print(path, files) if files and path not in self.paths: self.paths.append(path) # persistent on paths with models found model_files += files @@ -345,12 +345,12 @@ class ConfigPopup(QuickConfigGui): self.model_files.text = models_available_msg MODELS = { -# "pure 20b": "https://github.com/lightvector/KataGo/releases/download/v1.4.0/g170-b20c256x2-s4384473088-d968438914.bin.gz", -# "pure 30b": "https://github.com/lightvector/KataGo/releases/download/v1.4.0/g170-b30c320x2-s3530176512-d968463914.bin.gz", -# "pure 40b": "https://github.com/lightvector/KataGo/releases/download/v1.4.0/g170-b40c256x2-s3708042240-d967973220.bin.gz", - "final 20b":"https://github.com/lightvector/KataGo/releases/download/v1.4.5/g170e-b20c256x2-s5303129600-d1228401921.bin.gz", + # "pure 20b": "https://github.com/lightvector/KataGo/releases/download/v1.4.0/g170-b20c256x2-s4384473088-d968438914.bin.gz", + # "pure 30b": "https://github.com/lightvector/KataGo/releases/download/v1.4.0/g170-b30c320x2-s3530176512-d968463914.bin.gz", + # "pure 40b": "https://github.com/lightvector/KataGo/releases/download/v1.4.0/g170-b40c256x2-s3708042240-d967973220.bin.gz", + "final 20b": "https://github.com/lightvector/KataGo/releases/download/v1.4.5/g170e-b20c256x2-s5303129600-d1228401921.bin.gz", "final 30b": "https://github.com/lightvector/KataGo/releases/download/v1.4.5/g170-b30c320x2-s4824661760-d1229536699.bin.gz", - "final 40b":"https://github.com/lightvector/KataGo/releases/download/v1.4.5/g170-b40c256x2-s5095420928-d1229425124.bin.gz" + "final 40b": "https://github.com/lightvector/KataGo/releases/download/v1.4.5/g170-b40c256x2-s5095420928-d1229425124.bin.gz", } def download_models(self, *_largs):