REINVENT v3.2

Author: GuoJeff

Dockerfile

@@ -0,0 +1,34 @@
+FROM docker.io/continuumio/conda-ci-linux-64-python3.7:latest
+
+USER root
+
+RUN apt-get update && \
+    apt-get -y install rsync procps && \
+    wget https://sourceforge.net/projects/lmod/files/lua-5.1.4.9.tar.bz2 && \
+    tar xf lua-5.1.4.9.tar.bz2 && \
+    cd lua-5.1.4.9 && \
+    ./configure --prefix=/opt/apps/lua/5.1.4.9  && \
+    make; make install && \
+    cd /opt/apps/lua; ln -s 5.1.4.9 lua && \
+    ln -s /opt/apps/lua/lua/bin/lua /usr/local/bin && \
+    ln -s /opt/apps/lua/lua/bin/luac /usr/local/bin && \
+    cd; wget https://sourceforge.net/projects/lmod/files/Lmod-8.2.tar.bz2 && \
+    tar xf Lmod-8.2.tar.bz2 && \
+    cd Lmod-8.2; ./configure --prefix=/opt/apps --with-fastTCLInterp=no && \
+    make install && \
+    ln -s /opt/apps/lmod/lmod/init/profile /etc/profile.d/z00_lmod.sh
+
+ENV LMOD_ROOT=/opt/apps/lmod \
+    LMOD_PKG=/opt/apps/lmod/lmod \
+    LMOD_VERSION=8.2 \
+    LMOD_CMD=/opt/apps/lmod/lmod/libexec/lmod \
+    LMOD_DIR=/opt/apps/lmod/lmod/libexec \
+    BASH_ENV=/opt/apps/lmod/lmod/init/bash
+
+COPY . /reinventcli/
+
+WORKDIR /reinventcli
+
+RUN conda update -n base -c defaults conda && \
+    conda env update --name=base --file=reinvent.yml && \
+    chmod -R "a+rx" /reinventcli

README.md

@@ -1,4 +1,4 @@
-REINVENT 3.1
+REINVENT 3.2
 =================================================================================================================
 
 Installation
@@ -12,7 +12,7 @@ Installation
 
 4. Activate the environment:
 
-        $ conda activate reinvent.v3.0
+        $ conda activate reinvent.v3.2
 
 5. Use the tool.
 

configs/example.config.json

@@ -18,6 +18,13 @@
       "AZGARD_EXECUTOR_SCRIPT_PATH": "/<your_path>/executor.py",
       "AZGARD_ENV_PATH": "/<your_path>/miniconda3/envs/AZgard/bin/python",
       "AZGARD_DEBUG": true
+    },
+    "ICOLOS": {
+      "ICOLOS_EXECUTOR_PATH": "/<your_path>/miniconda3/envs/icolosprod/bin/icolos",
+      "ICOLOS_DEBUG": true
+    },
+    "AIZYNTH": {
+      "CONFIG": "/projects/mai/synthesisplanning/minimal_config.yml"
     }
   },
   "ENVIRONMENTAL_VARIABLES": {
@@ -28,5 +35,7 @@
   "ACTIVITY_CLASSIFICATION": "",
   "SMILES_SET_PATH": "",
   "PRIOR_PATH": "",
-  "LIBINVENT_PRIOR_PATH": ""
+  "LIBINVENT_PRIOR_PATH": "",
+  "SMILES_SET_LINK_INVENT_PATH":"",
+  "LINK_INVENT_PRIOR_PATH": ""
 }

input.py

@@ -3,19 +3,38 @@
 
 import sys
 import json
+import argparse
+from pathlib import Path
 from running_modes.manager import Manager
 
 
-if __name__ == "__main__":
+DEFAULT_BASE_CONFIG_PATH = (Path(__file__).parent / 'configs/config.json').resolve()
+
+parser = argparse.ArgumentParser(description='Run Reinvent.')
+parser.add_argument(
+    '--base_config', type=str, default=DEFAULT_BASE_CONFIG_PATH,
+    help='Path to basic configuration for Reinvent environment.'
+)
+parser.add_argument(
+    'run_config', type=str,
+    help='Path to configuration json file for this run.'
+)
 
-    with open(sys.argv[1]) as f:
-        json_input = f.read().replace('\r', '').replace('\n', '')
 
-    configuration = {}
+def read_json_file(path):
+    with open(path) as f:
+        json_input = f.read().replace('\r', '').replace('\n', '')
     try:
-        configuration = json.loads(json_input)
-    except (ValueError, KeyError, TypeError):
-        print("JSON format error")
-    else:
-        manager = Manager(configuration)
-        manager.run()
+        return json.loads(json_input)
+    except (ValueError, KeyError, TypeError) as e:
+        print(f"JSON format error in file ${path}: \n ${e}")
+
+
+if __name__ == "__main__":
+    args = parser.parse_args()
+
+    base_config = read_json_file(args.base_config)
+    run_config = read_json_file(args.run_config)
+
+    manager = Manager(base_config, run_config)
+    manager.run()

pytest.ini

@@ -1,4 +1,4 @@
-name: reinvent.v3.0
+name: reinvent.v3.2
 channels:
   - rdkit
   - pytorch
@@ -211,9 +211,9 @@ dependencies:
     - markdown==3.2.1
     - opt-einsum==3.2.0
     - protobuf==3.11.3
-    - reinvent-chemistry==0.0.40
-    - reinvent-models==0.0.12
-    - reinvent-scoring==0.0.57
+    - reinvent-chemistry==0.0.50
+    - reinvent-models==0.0.15rc1
+    - reinvent-scoring==0.0.73
     - tensorboard==1.15.0
     - tensorflow==1.15.2
     - tensorflow-estimator==1.15.1

reinvent.yml

@@ -0,0 +1,3 @@
+[Dolphin]
+Timestamp=2022,4,8,15,57,33
+Version=3

running_modes/.directory

@@ -0,0 +1,4 @@
+from running_modes.automated_curriculum_learning.actions.base_action import BaseAction
+from running_modes.automated_curriculum_learning.actions.base_sample_action import BaseSampleAction
+from running_modes.automated_curriculum_learning.actions.lib_invent_sample_model import LibInventSampleModel
+from running_modes.automated_curriculum_learning.actions.link_invent_sample_model import LinkInventSampleModel

running_modes/automated_curriculum_learning/actions/__init__.py

@@ -0,0 +1,11 @@
+import abc
+from running_modes.automated_curriculum_learning.logging.base_logger import BaseLogger
+
+
+class BaseAction(abc.ABC):
+    def __init__(self, logger=None):
+        """
+        (Abstract) Initializes an action.
+        :param logger: An optional logger instance.
+        """
+        self.logger: BaseLogger = logger

running_modes/automated_curriculum_learning/actions/base_action.py

@@ -0,0 +1,11 @@
+import numpy as np
+from running_modes.automated_curriculum_learning.actions import BaseAction
+
+
+class BaseSampleAction(BaseAction):
+
+    def _get_indices_of_unique_smiles(self, smiles: [str]) -> np.array:
+        """Returns an np.array of indices corresponding to the first entries in a list of smiles strings"""
+        _, idxs = np.unique(smiles, return_index=True)
+        sorted_indices = np.sort(idxs)
+        return sorted_indices