浏览代码

Use BufferedObserver for startup, bump CHIA version

Clemens-Alexander Brust 4 年之前
父节点
当前提交
047bb0b75b
共有 3 个文件被更改,包括 10 次插入4 次删除
  1. 8 2
      chillax/experiment_selfsupervised.py
  2. 1 1
      chillax/version.py
  3. 1 1
      setup.py

+ 8 - 2
chillax/experiment_selfsupervised.py

@@ -8,8 +8,11 @@ import argparse
 
 
 
 
 def main(config_files):
 def main(config_files):
-    # This shouldn't be necessary, but...
-    helpers.setup_environment()
+    # Set up buffered observer
+    buffered_observer = instrumentation.ObserverFactory.create({"name": "buffered"})
+    
+    # Set some important environment variables and validate the GPU configuration
+    helpers.setup_environment([buffered_observer])
 
 
     configs = [
     configs = [
         pcfg.config_from_json(config_file, read_from_file=True)
         pcfg.config_from_json(config_file, read_from_file=True)
@@ -24,6 +27,9 @@ def main(config_files):
     obs = instrumentation.NamedObservable("Experiment")
     obs = instrumentation.NamedObservable("Experiment")
 
 
     experiment_container = containers.ExperimentContainer(config, outer_observable=obs)
     experiment_container = containers.ExperimentContainer(config, outer_observable=obs)
+    
+    # Replay the buffer
+    buffered_observer.replay_messages(obs)
 
 
     with experiment_container.exception_shroud:
     with experiment_container.exception_shroud:
         obs.log_info("Hello!")
         obs.log_info("Hello!")

+ 1 - 1
chillax/version.py

@@ -1 +1 @@
-__version__ = "0.1a7"
+__version__ = "0.1a8"

+ 1 - 1
setup.py

@@ -17,7 +17,7 @@ setup(
     packages=find_packages(),
     packages=find_packages(),
     python_requires=">=3.7",
     python_requires=">=3.7",
     install_requires=[
     install_requires=[
-        "chia==2.0rc13",
+        "chia==2.0rc14",
     ],
     ],
     # metadata to display on PyPI
     # metadata to display on PyPI
     author="Clemens-Alexander Brust",
     author="Clemens-Alexander Brust",