composabl-cli-dev 0.19.1.dev1__cp310-cp310-macosx_11_0_arm64.whl

composabl_cli/__init__.py

@@ -0,0 +1,9 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+# We want to expose these classes to the user at the base of core
+# ex: `from composabl.core import Agent``
+
+# Continue with the imports
+from .cli import *

composabl_cli/template/agent_composabl/agent.py

@@ -0,0 +1,90 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+import os
+from typing import Dict
+
+from composabl import Agent, MaintainGoal, Sensor, Skill, Trainer
+
+# Accept the EULA
+os.environ["COMPOSABL_EULA_AGREED"] = "1"
+
+# Set the license
+# os.environ["COMPOSABL_LICENSE"] = "YOUR_LICENSE_KEY"
+
+
+class BalanceTeacher(MaintainGoal):
+    def __init__(self, *args, **kwargs):
+        super().__init__(
+            "pole_theta", "Maintain pole to upright", target=0, stop_distance=0.418
+        )
+
+        # defaults the BL and BR
+
+    async def compute_action_mask(self, transformed_sensors: Dict, action):
+        # Cartpole might not use an action mask, so this can return None
+        return None
+
+    async def transform_sensors(self, sensors, action):
+        # For Cartpole, might just return sensors directly
+        return sensors
+
+    async def transform_action(self, transformed_sensors: Dict, action):
+        # No transformation needed for discrete action space
+        return action
+
+    async def filtered_sensor_space(self):
+        # Return relevant sensors
+        return ["cart_pos", "cart_vel", "pole_theta", "pole_alpha"]
+
+
+def main():
+    # Create the agent
+    a = Agent()
+
+    # https://gymnasium.farama.org/environments/classic_control/cart_pole/#observation-space
+    a.add_sensors(
+        [
+            Sensor(
+                "cart_pos",
+                "The Cart Position between [-4.8, 4.8]",
+                lambda sensors: sensors[0],
+            ),
+            Sensor(
+                "cart_vel",
+                "The Cart Position between  [-inf, inf]",
+                lambda sensors: sensors[1],
+            ),
+            Sensor(
+                "pole_theta",
+                "The Pole Angle [-0.418 rad, 0.418 rad]",
+                lambda sensors: sensors[2],
+            ),
+            Sensor(
+                "pole_alpha",
+                "The Pole Angular Velocity [-inf, inf]",
+                lambda sensors: sensors[3],
+            ),
+        ]
+    )
+
+    skill = Skill("pole-balance", BalanceTeacher)
+    a.add_skill(skill)
+
+    # Create trainer and train the agent
+    r = Trainer(
+        {
+            "target": {"composabl"},
+            "resources": {
+                "sim_count": 2,
+            },
+        }
+    )
+
+    r.train(a, train_cycles=5)
+    r.close()
+
+
+if __name__ == "__main__":
+    main()

composabl_cli/template/agent_docker/agent.py

@@ -0,0 +1,99 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+import os
+from typing import Dict
+
+from composabl import Agent, MaintainGoal, Sensor, Skill, Trainer
+
+# Accept the EULA
+os.environ["COMPOSABL_EULA_AGREED"] = "1"
+
+# Set the license
+# os.environ["COMPOSABL_LICENSE"] = "YOUR_LICENSE_KEY"
+
+
+class BalanceTeacher(MaintainGoal):
+    def __init__(self, *args, **kwargs):
+        super().__init__(
+            "pole_theta", "Maintain pole to upright", target=0, stop_distance=0.418
+        )
+
+        # defaults the BL and BR
+
+    async def compute_action_mask(self, transformed_sensors: Dict, action):
+        # Cartpole might not use an action mask, so this can return None
+        return None
+
+    async def transform_sensors(self, sensors, action):
+        # For Cartpole, might just return sensors directly
+        return sensors
+
+    async def transform_action(self, transformed_sensors: Dict, action):
+        # No transformation needed for discrete action space
+        return action
+
+    async def filtered_sensor_space(self):
+        # Return relevant sensors
+        return ["cart_pos", "cart_vel", "pole_theta", "pole_alpha"]
+
+
+def main():
+    # Create the agent
+    a = Agent()
+
+    # https://gymnasium.farama.org/environments/classic_control/cart_pole/#observation-space
+    a.add_sensors(
+        [
+            Sensor(
+                "cart_pos",
+                "The Cart Position between [-4.8, 4.8]",
+                lambda sensors: sensors[0],
+            ),
+            Sensor(
+                "cart_vel",
+                "The Cart Position between  [-inf, inf]",
+                lambda sensors: sensors[1],
+            ),
+            Sensor(
+                "pole_theta",
+                "The Pole Angle [-0.418 rad, 0.418 rad]",
+                lambda sensors: sensors[2],
+            ),
+            Sensor(
+                "pole_alpha",
+                "The Pole Angular Velocity [-inf, inf]",
+                lambda sensors: sensors[3],
+            ),
+        ]
+    )
+
+    skill = Skill("pole-balance", BalanceTeacher)
+    a.add_skill(skill)
+
+    # Create trainer and train the agent
+    r = Trainer(
+        {
+            "target": {"docker": {"image": "composabl/sim-cartpole-dev:latest"}},
+            "resources": {
+                "sim_count": 2,
+            },
+            "post_processing": {
+                "record": {
+                    "is_enabled": True,
+                    "file_path": "/tmp/composabl/recordings",
+                    "avi_file_name": "output.avi",
+                    "gif_file_name": "output.gif",
+                    "max_frames": 24 * 5,
+                },
+            },
+        }
+    )
+
+    r.train(a, train_cycles=5)
+    r.close()
+
+
+if __name__ == "__main__":
+    main()

composabl_cli/template/agent_local/agent.py

@@ -0,0 +1,96 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+import os
+from typing import Dict
+
+from composabl import Agent, MaintainGoal, Sensor, Skill, Trainer
+
+# Accept the EULA
+os.environ["COMPOSABL_EULA_AGREED"] = "1"
+
+# Set the license
+# os.environ["COMPOSABL_LICENSE"] = "YOUR_LICENSE_KEY"
+
+
+class BalanceTeacher(MaintainGoal):
+    def __init__(self, *args, **kwargs):
+        super().__init__(
+            "pole_theta", "Maintain pole to upright", target=0, stop_distance=0.418
+        )
+
+        # defaults the BL and BR
+
+    async def compute_action_mask(self, transformed_sensors: Dict, action):
+        # Cartpole might not use an action mask, so this can return None
+        return None
+
+    async def transform_sensors(self, sensors, action):
+        # For Cartpole, might just return sensors directly
+        return sensors
+
+    async def transform_action(self, transformed_sensors: Dict, action):
+        # No transformation needed for discrete action space
+        return action
+
+    async def filtered_sensor_space(self):
+        # Return relevant sensors
+        return ["cart_pos", "cart_vel", "pole_theta", "pole_alpha"]
+
+
+def main():
+    # Create the agent
+    a = Agent()
+
+    # https://gymnasium.farama.org/environments/classic_control/cart_pole/#observation-space
+    a.add_sensors(
+        [
+            Sensor(
+                "cart_pos",
+                "The Cart Position between [-4.8, 4.8]",
+                lambda sensors: sensors[0],
+            ),
+            Sensor(
+                "cart_vel",
+                "The Cart Position between  [-inf, inf]",
+                lambda sensors: sensors[1],
+            ),
+            Sensor(
+                "pole_theta",
+                "The Pole Angle [-0.418 rad, 0.418 rad]",
+                lambda sensors: sensors[2],
+            ),
+            Sensor(
+                "pole_alpha",
+                "The Pole Angular Velocity [-inf, inf]",
+                lambda sensors: sensors[3],
+            ),
+        ]
+    )
+
+    skill = Skill("pole-balance", BalanceTeacher)
+    a.add_skill(skill)
+
+    # Create trainer and train the agent
+    r = Trainer(
+        {
+            "target": {"local": {"address": "localhost:1337"}},
+            "post_processing": {
+                "record": {
+                    "is_enabled": True,
+                    "file_path": "/tmp/composabl/recordings",
+                    "avi_file_name": "output.avi",
+                    "gif_file_name": "output.gif",
+                    "max_frames": 24 * 5,
+                },
+            },
+        }
+    )
+
+    r.train(a, train_cycles=5)
+    r.close()
+
+
+if __name__ == "__main__":
+    main()

composabl_cli/template/perceptor/README.md

@@ -0,0 +1,67 @@
+# README
+
+This is a template for creating a new Perceptor. A Perceptor is a component that is responsible for processing data and returning a result. This can be used for example in a Machine Learning model, where the Perceptor is responsible for processing the data and returning the prediction.
+
+## Tree Structure
+
+The template is structured as follows:
+
+```bash
+my-perceptor/           # Root folder
+├── my_perceptor/       # Main package folder
+│   ├── __init__.py     # Package init file
+│   └── perceptor.py    # Main perceptor file
+├── pyproject.toml      # Project configuration, containing [composabl]
+```
+
+## PyProject [composabl] Section
+
+We add the `[composabl]` section to the `pyproject.toml` file to specify the type of component we are creating as well as its entrypoint. This is used by the Composabl CLI to determine the type of
+component and how to handle it.
+
+Example:
+
+```
+[composabl]
+type = "teacher"
+entrypoint = "my_perceptor.perceptor:MyPerceptor"
+```
+
+## Development
+
+To work on the Perceptor, you can simply create a temporary file or main file that starts up and executes the `compute` method of the portable Perceptor. Example, we can create a `test.py` file with:
+
+```python
+from composabl_perceptor_my_perceptor.perceptor import MyPerceptor
+
+
+async def start():
+    p = MyPerceptor()
+    res = await t.compute(None, [1.0])
+    print(res)
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(start())
+```
+
+Which we can then run with
+
+```bash
+# Install the module
+pip install -e my-perceptor
+
+# Run the test file
+python my-perceptor/test.py
+```
+
+### Preparing for Upload
+
+Once we are ready for uploading, we can create a `.tar.gz` file that contains the version. This can be done with the following command:
+
+```bash
+# Tar GZ the plugin
+tar -czvf my-perceptor-0.0.1.tar.gz my-perceptor
+```

composabl_cli/template/perceptor/pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+name = "{{NAME}}"
+version = "0.1.0"
+description = "{{DESCRIPTION}}"
+authors = [{ name = "John Doe", email = "john.doe@composabl.com" }]
+dependencies = [
+    "composabl-core{{COMPOSABL_DEV}}{{COMPOSABL_CORE_VERSION}}",
+]
+
+[composabl]
+type = "perceptor"
+variables = [
+    "counter_derived"
+]
+entrypoint = "{{NAME_MODULE}}.perceptor:DemoPerceptor"

composabl_cli/template/perceptor/{{NAME_MODULE}}/__init__.py

@@ -0,0 +1,3 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential

composabl_cli/template/perceptor/{{NAME_MODULE}}/perceptor.py

@@ -0,0 +1,34 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+import time
+
+from composabl_core import PerceptorImpl
+
+
+class DemoPerceptor(PerceptorImpl):
+    def __init__(self, *args, **kwargs):
+        self.previous_value = None
+        self.previous_time = None
+
+    async def compute(self, obs_spec, obs):
+        current_value = obs["counter"]
+        current_time = time.time()  # Get current time
+        value_derived = 0
+
+        # Only calculate derivative if we have a previous sensors values
+        if self.previous_value is not None and self.previous_time is not None:
+            time_delta = current_time - self.previous_time
+
+            if time_delta > 0:  # Check to ensure time has passed
+                value_derived = (current_value - self.previous_value) / time_delta
+
+        # Update previous value and time for the next sensor values
+        self.previous_value = current_value
+        self.previous_time = current_time
+
+        return {"counter_derived": value_derived}
+
+    def filtered_sensor_space(self, obs):
+        return ["counter"]

composabl_cli/template/selector_controller/README.md

@@ -0,0 +1,67 @@
+# README
+
+This is a template for creating a new Perceptor. A Perceptor is a component that is responsible for processing data and returning a result. This can be used for example in a Machine Learning model, where the Perceptor is responsible for processing the data and returning the prediction.
+
+## Tree Structure
+
+The template is structured as follows:
+
+```bash
+my-perceptor/           # Root folder
+├── my_perceptor/       # Main package folder
+│   ├── __init__.py     # Package init file
+│   └── perceptor.py    # Main perceptor file
+├── pyproject.toml      # Project configuration, containing [composabl]
+```
+
+## PyProject [composabl] Section
+
+We add the `[composabl]` section to the `pyproject.toml` file to specify the type of component we are creating as well as its entrypoint. This is used by the Composabl CLI to determine the type of
+component and how to handle it.
+
+Example:
+
+```
+[composabl]
+type = "teacher"
+entrypoint = "my_perceptor.perceptor:MyPerceptor"
+```
+
+## Development
+
+To work on the Perceptor, you can simply create a temporary file or main file that starts up and executes the `compute` method of the portable Perceptor. Example, we can create a `test.py` file with:
+
+```python
+from composabl_perceptor_my_perceptor.perceptor import MyPerceptor
+
+
+async def start():
+    p = MyPerceptor()
+    res = await t.compute(None, [1.0])
+    print(res)
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(start())
+```
+
+Which we can then run with
+
+```bash
+# Install the module
+pip install -e my-perceptor
+
+# Run the test file
+python my-perceptor/test.py
+```
+
+### Preparing for Upload
+
+Once we are ready for uploading, we can create a `.tar.gz` file that contains the version. This can be done with the following command:
+
+```bash
+# Tar GZ the plugin
+tar -czvf my-perceptor-0.0.1.tar.gz my-perceptor
+```

composabl_cli/template/selector_controller/pyproject.toml

@@ -0,0 +1,12 @@
+[project]
+name = "{{NAME}}"
+version = "0.1.0"
+description = "{{DESCRIPTION}}"
+authors = [{ name = "John Doe", email = "john.doe@composabl.com" }]
+dependencies = [
+    "composabl-core{{COMPOSABL_DEV}}{{COMPOSABL_CORE_VERSION}}",
+]
+
+[composabl]
+type = "selector-controller"
+entrypoint = "{{NAME_MODULE}}.controller:Controller"

composabl_cli/template/selector_controller/{{NAME_MODULE}}/__init__.py

@@ -0,0 +1,3 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential

composabl_cli/template/selector_controller/{{NAME_MODULE}}/controller.py

@@ -0,0 +1,52 @@
+# Copyright (C) Composabl, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+from typing import Dict, List
+
+from composabl_core import SkillController
+
+
+class Controller(SkillController):
+    """
+    We start at 10 reward and count down to 0 the goal is that the agent stays
+    above or equal to 0 this means it learned to cound +1 each time
+    """
+
+    def __init__(self, *args, **kwargs):
+        self.past_obs = None
+        self.counter = 10
+        self.sensor_name = "counter"
+
+    async def compute_action(self, transformed_sensors: Dict, action) -> List[float]:
+        """
+        Compute action in a programme selector is to select the best skill based in a rule or an optimization.
+        Let's supose that we have three skills:
+        1 - Decrease counter
+        2 - Increase counter
+        3 - Stop
+        """
+        # return Skill 1 = decrease
+        if self.counter > 10:
+            return [0]
+        # return Skill 2 = increase
+        elif self.counter < 10:
+            return [1]
+        # return Skill 3 = stop
+        else:
+            return [3]
+
+    async def compute_success_criteria(self, transformed_sensors: Dict, action) -> bool:
+        return bool(transformed_sensors[self.sensor_name] >= 10)
+
+    async def compute_termination(self, transformed_sensors: Dict, action) -> bool:
+        return bool(transformed_sensors[self.sensor_name] <= -10)
+
+    async def transform_sensors(self, sensors, action) -> str:
+        return sensors
+
+    async def transform_action(self, transformed_sensors: Dict, action) -> float:
+        return action
+
+    async def filtered_sensor_space(self) -> List[str]:
+        return ["counter"]

composabl_cli/template/selector_teacher/README.md

@@ -0,0 +1,67 @@
+# README
+
+This is a template for creating a new Perceptor. A Perceptor is a component that is responsible for processing data and returning a result. This can be used for example in a Machine Learning model, where the Perceptor is responsible for processing the data and returning the prediction.
+
+## Tree Structure
+
+The template is structured as follows:
+
+```bash
+my-perceptor/           # Root folder
+├── my_perceptor/       # Main package folder
+│   ├── __init__.py     # Package init file
+│   └── perceptor.py    # Main perceptor file
+├── pyproject.toml      # Project configuration, containing [composabl]
+```
+
+## PyProject [composabl] Section
+
+We add the `[composabl]` section to the `pyproject.toml` file to specify the type of component we are creating as well as its entrypoint. This is used by the Composabl CLI to determine the type of
+component and how to handle it.
+
+Example:
+
+```
+[composabl]
+type = "teacher"
+entrypoint = "my_perceptor.perceptor:MyPerceptor"
+```
+
+## Development
+
+To work on the Perceptor, you can simply create a temporary file or main file that starts up and executes the `compute` method of the portable Perceptor. Example, we can create a `test.py` file with:
+
+```python
+from composabl_perceptor_my_perceptor.perceptor import MyPerceptor
+
+
+async def start():
+    p = MyPerceptor()
+    res = await t.compute(None, [1.0])
+    print(res)
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(start())
+```
+
+Which we can then run with
+
+```bash
+# Install the module
+pip install -e my-perceptor
+
+# Run the test file
+python my-perceptor/test.py
+```
+
+### Preparing for Upload
+
+Once we are ready for uploading, we can create a `.tar.gz` file that contains the version. This can be done with the following command:
+
+```bash
+# Tar GZ the plugin
+tar -czvf my-perceptor-0.0.1.tar.gz my-perceptor
+```