luckyrobots 0.1.69__py3-none-any.whl → 0.1.71__py3-none-any.whl

luckyrobots/grpc/generated/viewport_pb2.py

@@ -9,9 +9,13 @@ from google.protobuf import descriptor_pool as _descriptor_pool
 from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
-
 _runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC, 6, 31, 1, "", "viewport.proto"
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'viewport.proto'
 )
 # @@protoc_insertion_point(imports)
 
@@ -21,28 +25,26 @@ _sym_db = _symbol_database.Default()
 from . import media_pb2 as media__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x0eviewport.proto\x12\x0chazel.rpc.v1\x1a\x0bmedia.proto"v\n\x1aStartViewportStreamRequest\x12\x15\n\rviewport_name\x18\x01 \x01(\t\x12\x12\n\ntarget_fps\x18\x02 \x01(\r\x12\r\n\x05width\x18\x03 \x01(\r\x12\x0e\n\x06height\x18\x04 \x01(\r\x12\x0e\n\x06\x66ormat\x18\x05 \x01(\t"l\n\x14ViewportStreamConfig\x12\x11\n\tstreaming\x18\x01 \x01(\x08\x12\x15\n\rviewport_name\x18\x02 \x01(\t\x12\x0b\n\x03\x66ps\x18\x03 \x01(\r\x12\r\n\x05width\x18\x04 \x01(\r\x12\x0e\n\x06height\x18\x05 \x01(\r"\x1b\n\x19StopViewportStreamRequest"-\n\x1aStopViewportStreamResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08"\x18\n\x16GetViewportInfoRequest"r\n\x17GetViewportInfoResponse\x12\x1b\n\x13\x61vailable_viewports\x18\x01 \x03(\t\x12:\n\x0e\x63urrent_config\x18\x02 \x01(\x0b\x32".hazel.rpc.v1.ViewportStreamConfig2\xc9\x01\n\x0fViewportService\x12^\n\x0fGetViewportInfo\x12$.hazel.rpc.v1.GetViewportInfoRequest\x1a%.hazel.rpc.v1.GetViewportInfoResponse\x12V\n\x0eStreamViewport\x12(.hazel.rpc.v1.StartViewportStreamRequest\x1a\x18.hazel.rpc.v1.ImageFrame0\x01\x42\x03\xf8\x01\x01\x62\x06proto3'
-)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0eviewport.proto\x12\thazel.rpc\x1a\x0bmedia.proto\"v\n\x1aStartViewportStreamRequest\x12\x15\n\rviewport_name\x18\x01 \x01(\t\x12\x12\n\ntarget_fps\x18\x02 \x01(\r\x12\r\n\x05width\x18\x03 \x01(\r\x12\x0e\n\x06height\x18\x04 \x01(\r\x12\x0e\n\x06\x66ormat\x18\x05 \x01(\t\"l\n\x14ViewportStreamConfig\x12\x11\n\tstreaming\x18\x01 \x01(\x08\x12\x15\n\rviewport_name\x18\x02 \x01(\t\x12\x0b\n\x03\x66ps\x18\x03 \x01(\r\x12\r\n\x05width\x18\x04 \x01(\r\x12\x0e\n\x06height\x18\x05 \x01(\r\"\x1b\n\x19StopViewportStreamRequest\"-\n\x1aStopViewportStreamResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\"\x18\n\x16GetViewportInfoRequest\"o\n\x17GetViewportInfoResponse\x12\x1b\n\x13\x61vailable_viewports\x18\x01 \x03(\t\x12\x37\n\x0e\x63urrent_config\x18\x02 \x01(\x0b\x32\x1f.hazel.rpc.ViewportStreamConfig2\xbd\x01\n\x0fViewportService\x12X\n\x0fGetViewportInfo\x12!.hazel.rpc.GetViewportInfoRequest\x1a\".hazel.rpc.GetViewportInfoResponse\x12P\n\x0eStreamViewport\x12%.hazel.rpc.StartViewportStreamRequest\x1a\x15.hazel.rpc.ImageFrame0\x01\x42\x03\xf8\x01\x01\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "viewport_pb2", _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'viewport_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
-    _globals["DESCRIPTOR"]._loaded_options = None
-    _globals["DESCRIPTOR"]._serialized_options = b"\370\001\001"
-    _globals["_STARTVIEWPORTSTREAMREQUEST"]._serialized_start = 45
-    _globals["_STARTVIEWPORTSTREAMREQUEST"]._serialized_end = 163
-    _globals["_VIEWPORTSTREAMCONFIG"]._serialized_start = 165
-    _globals["_VIEWPORTSTREAMCONFIG"]._serialized_end = 273
-    _globals["_STOPVIEWPORTSTREAMREQUEST"]._serialized_start = 275
-    _globals["_STOPVIEWPORTSTREAMREQUEST"]._serialized_end = 302
-    _globals["_STOPVIEWPORTSTREAMRESPONSE"]._serialized_start = 304
-    _globals["_STOPVIEWPORTSTREAMRESPONSE"]._serialized_end = 349
-    _globals["_GETVIEWPORTINFOREQUEST"]._serialized_start = 351
-    _globals["_GETVIEWPORTINFOREQUEST"]._serialized_end = 375
-    _globals["_GETVIEWPORTINFORESPONSE"]._serialized_start = 377
-    _globals["_GETVIEWPORTINFORESPONSE"]._serialized_end = 491
-    _globals["_VIEWPORTSERVICE"]._serialized_start = 494
-    _globals["_VIEWPORTSERVICE"]._serialized_end = 695
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\370\001\001'
+  _globals['_STARTVIEWPORTSTREAMREQUEST']._serialized_start=42
+  _globals['_STARTVIEWPORTSTREAMREQUEST']._serialized_end=160
+  _globals['_VIEWPORTSTREAMCONFIG']._serialized_start=162
+  _globals['_VIEWPORTSTREAMCONFIG']._serialized_end=270
+  _globals['_STOPVIEWPORTSTREAMREQUEST']._serialized_start=272
+  _globals['_STOPVIEWPORTSTREAMREQUEST']._serialized_end=299
+  _globals['_STOPVIEWPORTSTREAMRESPONSE']._serialized_start=301
+  _globals['_STOPVIEWPORTSTREAMRESPONSE']._serialized_end=346
+  _globals['_GETVIEWPORTINFOREQUEST']._serialized_start=348
+  _globals['_GETVIEWPORTINFOREQUEST']._serialized_end=372
+  _globals['_GETVIEWPORTINFORESPONSE']._serialized_start=374
+  _globals['_GETVIEWPORTINFORESPONSE']._serialized_end=485
+  _globals['_VIEWPORTSERVICE']._serialized_start=488
+  _globals['_VIEWPORTSERVICE']._serialized_end=677
 # @@protoc_insertion_point(module_scope)

luckyrobots/grpc/generated/viewport_pb2_grpc.py

@@ -6,31 +6,29 @@ import warnings
 from . import media_pb2 as media__pb2
 from . import viewport_pb2 as viewport__pb2
 
-GRPC_GENERATED_VERSION = "1.76.0"
+GRPC_GENERATED_VERSION = '1.76.0'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 
 try:
     from grpc._utilities import first_version_is_lower
-
-    _version_not_supported = first_version_is_lower(
-        GRPC_VERSION, GRPC_GENERATED_VERSION
-    )
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
 except ImportError:
     _version_not_supported = True
 
 if _version_not_supported:
     raise RuntimeError(
-        f"The grpc package installed is at version {GRPC_VERSION},"
-        + " but the generated code in viewport_pb2_grpc.py depends on"
-        + f" grpcio>={GRPC_GENERATED_VERSION}."
-        + f" Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}"
-        + f" or downgrade your generated code using grpcio-tools<={GRPC_VERSION}."
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + ' but the generated code in viewport_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
     )
 
 
 class ViewportServiceStub(object):
-    """Streams pixels from a named viewport (useful for external visualization / training input)."""
+    """Streams pixels from a named viewport (useful for external visualization / training input).
+    """
 
     def __init__(self, channel):
         """Constructor.
@@ -39,78 +37,73 @@ class ViewportServiceStub(object):
             channel: A grpc.Channel.
         """
         self.GetViewportInfo = channel.unary_unary(
-            "/hazel.rpc.v1.ViewportService/GetViewportInfo",
-            request_serializer=viewport__pb2.GetViewportInfoRequest.SerializeToString,
-            response_deserializer=viewport__pb2.GetViewportInfoResponse.FromString,
-            _registered_method=True,
-        )
+                '/hazel.rpc.ViewportService/GetViewportInfo',
+                request_serializer=viewport__pb2.GetViewportInfoRequest.SerializeToString,
+                response_deserializer=viewport__pb2.GetViewportInfoResponse.FromString,
+                _registered_method=True)
         self.StreamViewport = channel.unary_stream(
-            "/hazel.rpc.v1.ViewportService/StreamViewport",
-            request_serializer=viewport__pb2.StartViewportStreamRequest.SerializeToString,
-            response_deserializer=media__pb2.ImageFrame.FromString,
-            _registered_method=True,
-        )
+                '/hazel.rpc.ViewportService/StreamViewport',
+                request_serializer=viewport__pb2.StartViewportStreamRequest.SerializeToString,
+                response_deserializer=media__pb2.ImageFrame.FromString,
+                _registered_method=True)
 
 
 class ViewportServiceServicer(object):
-    """Streams pixels from a named viewport (useful for external visualization / training input)."""
+    """Streams pixels from a named viewport (useful for external visualization / training input).
+    """
 
     def GetViewportInfo(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
 
     def StreamViewport(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
 
 
 def add_ViewportServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
-        "GetViewportInfo": grpc.unary_unary_rpc_method_handler(
-            servicer.GetViewportInfo,
-            request_deserializer=viewport__pb2.GetViewportInfoRequest.FromString,
-            response_serializer=viewport__pb2.GetViewportInfoResponse.SerializeToString,
-        ),
-        "StreamViewport": grpc.unary_stream_rpc_method_handler(
-            servicer.StreamViewport,
-            request_deserializer=viewport__pb2.StartViewportStreamRequest.FromString,
-            response_serializer=media__pb2.ImageFrame.SerializeToString,
-        ),
+            'GetViewportInfo': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetViewportInfo,
+                    request_deserializer=viewport__pb2.GetViewportInfoRequest.FromString,
+                    response_serializer=viewport__pb2.GetViewportInfoResponse.SerializeToString,
+            ),
+            'StreamViewport': grpc.unary_stream_rpc_method_handler(
+                    servicer.StreamViewport,
+                    request_deserializer=viewport__pb2.StartViewportStreamRequest.FromString,
+                    response_serializer=media__pb2.ImageFrame.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
-        "hazel.rpc.v1.ViewportService", rpc_method_handlers
-    )
+            'hazel.rpc.ViewportService', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
-    server.add_registered_method_handlers(
-        "hazel.rpc.v1.ViewportService", rpc_method_handlers
-    )
+    server.add_registered_method_handlers('hazel.rpc.ViewportService', rpc_method_handlers)
 
 
-# This class is part of an EXPERIMENTAL API.
+ # This class is part of an EXPERIMENTAL API.
 class ViewportService(object):
-    """Streams pixels from a named viewport (useful for external visualization / training input)."""
+    """Streams pixels from a named viewport (useful for external visualization / training input).
+    """
 
     @staticmethod
-    def GetViewportInfo(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
+    def GetViewportInfo(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
         return grpc.experimental.unary_unary(
             request,
             target,
-            "/hazel.rpc.v1.ViewportService/GetViewportInfo",
+            '/hazel.rpc.ViewportService/GetViewportInfo',
             viewport__pb2.GetViewportInfoRequest.SerializeToString,
             viewport__pb2.GetViewportInfoResponse.FromString,
             options,
@@ -121,26 +114,23 @@ class ViewportService(object):
             wait_for_ready,
             timeout,
             metadata,
-            _registered_method=True,
-        )
+            _registered_method=True)
 
     @staticmethod
-    def StreamViewport(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
+    def StreamViewport(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
         return grpc.experimental.unary_stream(
             request,
             target,
-            "/hazel.rpc.v1.ViewportService/StreamViewport",
+            '/hazel.rpc.ViewportService/StreamViewport',
             viewport__pb2.StartViewportStreamRequest.SerializeToString,
             media__pb2.ImageFrame.FromString,
             options,
@@ -151,5 +141,4 @@ class ViewportService(object):
             wait_for_ready,
             timeout,
             metadata,
-            _registered_method=True,
-        )
+            _registered_method=True)

luckyrobots/grpc/proto/agent.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // Agent + observation service definitions for the LuckyEngine / Hazel ScriptCore gRPC API (v1).
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 
@@ -37,10 +37,42 @@ message StreamAgentRequest {
     uint32 target_fps = 2;
 }
 
+// Domain randomization configuration for physics parameters.
+// All range fields are [min, max] pairs. Empty ranges mean "use defaults".
+message DomainRandomizationConfig {
+    // Initial state randomization
+    repeated float pose_position_noise = 1;   // [x, y, z] position noise std
+    float pose_orientation_noise = 2;         // Orientation noise std (radians)
+    float joint_position_noise = 3;           // Joint position noise std
+    float joint_velocity_noise = 4;           // Joint velocity noise std
+
+    // Physics parameters (all [min, max] ranges)
+    repeated float friction_range = 5;        // Surface friction coefficient
+    repeated float restitution_range = 6;     // Bounce/restitution coefficient
+    repeated float mass_scale_range = 7;      // Body mass multiplier
+    repeated float com_offset_range = 8;      // Center of mass offset
+
+    // Motor/actuator randomization
+    repeated float motor_strength_range = 9;  // Motor strength multiplier
+    repeated float motor_offset_range = 10;   // Motor position offset
+
+    // External disturbances
+    repeated float push_interval_range = 11;  // Time between pushes [min, max]
+    repeated float push_velocity_range = 12;  // Push velocity magnitude [min, max]
+
+    // Terrain configuration
+    string terrain_type = 13;                 // Terrain type identifier
+    float terrain_difficulty = 14;            // Terrain difficulty level
+}
+
 message ResetAgentRequest {
     // Agent logical name. Convention in this repo is `agent_0`, `agent_1`, ...
     // Empty typically means "default agent" (agent_0).
     string agent_name = 1;
+
+    // Optional domain randomization config for this reset.
+    // If not provided, use defaults or keep previous values.
+    DomainRandomizationConfig dr_config = 2;
 }
 
 message ResetAgentResponse {

luckyrobots/grpc/proto/camera.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // Camera service definitions for the LuckyEngine / Hazel ScriptCore gRPC API (v1).
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/common.proto

@@ -5,7 +5,7 @@ syntax = "proto3";
 // Compatibility note:
 // - Avoid renumbering existing field tags (the `= N` numbers). Add new fields with new tags instead.
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/hazel_rpc.proto

@@ -15,7 +15,7 @@ syntax = "proto3";
 // Compatibility note:
 // - Avoid renumbering existing field tags (the `= N` numbers). Add new fields with new tags instead.
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/media.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // Media / image payload types shared across services.
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/mujoco.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // MuJoCo service definitions for the LuckyEngine / Hazel ScriptCore gRPC API (v1).
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/scene.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // Scene service definitions for the LuckyEngine / Hazel ScriptCore gRPC API (v1).
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/telemetry.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // Telemetry service definitions for the LuckyEngine / Hazel ScriptCore gRPC API (v1).
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/grpc/proto/viewport.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 // Viewport service definitions for the LuckyEngine / Hazel ScriptCore gRPC API (v1).
 
-package hazel.rpc.v1;
+package hazel.rpc;
 
 option cc_enable_arenas = true;
 

luckyrobots/luckyrobots.py

@@ -4,6 +4,8 @@ import time
 from collections.abc import Sequence
 from typing import Optional
 
+from typing import Any
+
 from .engine import launch_luckyengine, stop_luckyengine
 from .models import ObservationResponse
 from .client import LuckyEngineClient, GrpcConnectionError
@@ -170,6 +172,32 @@ class LuckyRobots:
             time.sleep(sleep_s)
         return self.get_observation()
 
+    def reset(
+        self,
+        agent_name: str = "",
+        randomization_cfg: Optional[Any] = None,
+    ) -> ObservationResponse:
+        """
+        Reset the agent and return a fresh observation.
+
+        Args:
+            agent_name: Agent logical name. Empty string means default agent.
+            randomization_cfg: Optional domain randomization config for this reset.
+                Use this to randomize physics parameters (friction, mass, etc.)
+                at the start of each episode for sim-to-real transfer.
+
+        Returns:
+            ObservationResponse after reset.
+
+        Raises:
+            RuntimeError: If reset fails.
+        """
+        client = self._require_client()
+        resp = client.reset_agent(agent_name=agent_name, randomization_cfg=randomization_cfg)
+        if hasattr(resp, "success") and not resp.success:
+            raise RuntimeError(f"Reset failed: {getattr(resp, 'message', '')}")
+        return self.get_observation(agent_name=agent_name)
+
     def close(self, stop_engine: bool = True) -> None:
         """Close gRPC client and optionally stop the engine executable."""
         if self._engine_client is not None:

luckyrobots/models/__init__.py

@@ -4,10 +4,12 @@ Pydantic models for LuckyRobots.
 
 from .observation import ObservationResponse, StateSnapshot
 from .camera import CameraData, CameraShape
+from .randomization import DomainRandomizationConfig
 
 __all__ = [
     "ObservationResponse",
     "StateSnapshot",
     "CameraData",
     "CameraShape",
+    "DomainRandomizationConfig",
 ]

luckyrobots/models/randomization.py

@@ -0,0 +1,77 @@
+"""
+Domain randomization configuration for physics parameters.
+
+This model maps to the DomainRandomizationConfig proto message in agent.proto.
+"""
+
+from typing import Optional
+from pydantic import BaseModel, Field, ConfigDict
+
+
+class DomainRandomizationConfig(BaseModel):
+    """Domain randomization configuration for physics parameters.
+
+    All range fields are [min, max] tuples. None/empty means "use defaults".
+
+    Usage:
+        randomization_cfg = DomainRandomizationConfig(
+            friction_range=(0.5, 1.5),
+            mass_scale_range=(0.8, 1.2),
+            joint_position_noise=0.05,
+        )
+        client.reset_agent(randomization_cfg=randomization_cfg)
+    """
+
+    model_config = ConfigDict(frozen=True)
+
+    # Initial state randomization
+    pose_position_noise: Optional[tuple[float, float, float]] = Field(
+        default=None, description="[x, y, z] position noise std"
+    )
+    pose_orientation_noise: Optional[float] = Field(
+        default=None, description="Orientation noise std (radians)"
+    )
+    joint_position_noise: Optional[float] = Field(
+        default=None, description="Joint position noise std"
+    )
+    joint_velocity_noise: Optional[float] = Field(
+        default=None, description="Joint velocity noise std"
+    )
+
+    # Physics parameters (all [min, max] ranges)
+    friction_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Surface friction coefficient [min, max]"
+    )
+    restitution_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Bounce/restitution coefficient [min, max]"
+    )
+    mass_scale_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Body mass multiplier [min, max]"
+    )
+    com_offset_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Center of mass offset [min, max]"
+    )
+
+    # Motor/actuator randomization
+    motor_strength_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Motor strength multiplier [min, max]"
+    )
+    motor_offset_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Motor position offset [min, max]"
+    )
+
+    # External disturbances
+    push_interval_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Time between pushes [min, max]"
+    )
+    push_velocity_range: Optional[tuple[float, float]] = Field(
+        default=None, description="Push velocity magnitude [min, max]"
+    )
+
+    # Terrain configuration
+    terrain_type: Optional[str] = Field(
+        default=None, description="Terrain type identifier"
+    )
+    terrain_difficulty: Optional[float] = Field(
+        default=None, description="Terrain difficulty level"
+    )

{luckyrobots-0.1.69.dist-info → luckyrobots-0.1.71.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: luckyrobots
-Version: 0.1.69
+Version: 0.1.71
 Summary: Robotics-AI Training in Hyperrealistic Game Environments
 Project-URL: Homepage, https://github.com/luckyrobots/luckyrobots
 Project-URL: Documentation, https://luckyrobots.readthedocs.io

luckyrobots-0.1.71.dist-info/RECORD

@@ -0,0 +1,47 @@
+luckyrobots/__init__.py,sha256=VWewiYhYbYCs1ZSD0KxozmG2jH2uLlBxsVgFIRivKTM,824
+luckyrobots/client.py,sha256=0i8lVaLHjXXXzNZH3DWoc3ViqWBeoInrPP9nefkNWR4,32646
+luckyrobots/luckyrobots.py,sha256=YLx8mvuOOmY-ZaAve_k2md0HIGG84__5oXUXF1ymjhI,7812
+luckyrobots/utils.py,sha256=62BqpSRBFykVRMZ5Ti2F-yKRDoyU3UNHeesDF_PVCFc,2958
+luckyrobots/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+luckyrobots/config/robots.yaml,sha256=Wu2AgwhZxo9F-RLcqPJXHnqzjT8a0wR-rFBcecck_eE,5164
+luckyrobots/engine/__init__.py,sha256=-3r7aLueDF7eyumqIhTvYqS8ZNnVSJUJEHG4ybsKrjc,546
+luckyrobots/engine/check_updates.py,sha256=KGQHrj1LB0_vexktO01xx-ZvPOYKkto4VBb3H1KV7uI,7593
+luckyrobots/engine/download.py,sha256=vIb5T40Jk2BnfjPo0f4KgW14Om6P4_mvPpZ9JAlkVvg,4029
+luckyrobots/engine/manager.py,sha256=8vxSevaoXo-lw2G6y74lVc-Ms8xIELK0yLswydMiOQY,13388
+luckyrobots/grpc/__init__.py,sha256=hPh-X_FEus2sxrfQB2fQjD754X5bMaUFTWm3mZmUhe0,153
+luckyrobots/grpc/generated/__init__.py,sha256=u69M4E_2LwRQSD_n4wvQyMX7dryBCrIdsho59c52Crk,732
+luckyrobots/grpc/generated/agent_pb2.py,sha256=me4OtOKGfwg5JUho6I5NV6v7ie-DO7T6iSCs8LcQ6yM,6555
+luckyrobots/grpc/generated/agent_pb2_grpc.py,sha256=VnjWZTKOE4KWRsy1n-ssfAk1kEP5uUqOhakWKzFtr4g,9123
+luckyrobots/grpc/generated/camera_pb2.py,sha256=2UDkgzMYdDZ53euPfLAndeI7k4wI9QH_Dvwkay6TAB4,2623
+luckyrobots/grpc/generated/camera_pb2_grpc.py,sha256=hUXzsf2qCE-ffZzkF8El2UuuLu6nxjKSYPukId2uFP8,5278
+luckyrobots/grpc/generated/common_pb2.py,sha256=0eZTxqlsZdTrgp2hgLy-hdDNtac5U3kkS_D9UbVy0e8,2045
+luckyrobots/grpc/generated/common_pb2_grpc.py,sha256=MWix04c710srn9BtjDxa9-UkZz0wa-w8tB0EmYndSgg,886
+luckyrobots/grpc/generated/hazel_rpc_pb2.py,sha256=XB1nGSVTYhpulRpS8Ja44KaCV5s-l1t_FPV4DJ5_ak8,1716
+luckyrobots/grpc/generated/hazel_rpc_pb2_grpc.py,sha256=5bhVh5DmVDl2ZsdgN7mTnElDbgU142TOI1qm8a8eblw,889
+luckyrobots/grpc/generated/media_pb2.py,sha256=TtGdBub2Pj1vG_jJJIZnIZealLD0s98wZ4KgGIrcn2c,1827
+luckyrobots/grpc/generated/media_pb2_grpc.py,sha256=szeOZzqDnIBFEyjAtXEHNxytTb9hTvOkQj0PhvXumMM,885
+luckyrobots/grpc/generated/mujoco_pb2.py,sha256=Oq1AdZ0h-svs7ntlaDZ-zvQdTPsulnaE7YaINvBIBpM,3476
+luckyrobots/grpc/generated/mujoco_pb2_grpc.py,sha256=Rz8hyPvhB2nR-0Mej4Ly9ixbrdpPCxDafmL8H4x43Nk,8603
+luckyrobots/grpc/generated/scene_pb2.py,sha256=LzCApFBj0RXsFhvLYW07VXbuZubaiACj6rX8QwbGSG4,3981
+luckyrobots/grpc/generated/scene_pb2_grpc.py,sha256=A-aj6kPPydehlq1PhspITymVNktK0k3fd7xDKMU0Zy0,8569
+luckyrobots/grpc/generated/telemetry_pb2.py,sha256=LgVtxJv7ovtqlrgxG375aXADGt3qXwgpsIhIkJ3pLKE,2790
+luckyrobots/grpc/generated/telemetry_pb2_grpc.py,sha256=8TuCSc7rnQZFNqhEPXUoJsWAnwvEvyyEI5GaqQUCSSs,5463
+luckyrobots/grpc/generated/viewport_pb2.py,sha256=Z_5gBD23z7He_OACYkaIsi0o4LIESzOG7o_rO1lRLVs,3093
+luckyrobots/grpc/generated/viewport_pb2_grpc.py,sha256=BkJqREcjIa3ZdNgZUxH_MvBWOte9VJJGi-KlFgYepog,5436
+luckyrobots/grpc/proto/agent.proto,sha256=XOyjtsTCU9iIl-cXCWVOWynF-WTSDaMVY7EwVEDMEsE,7282
+luckyrobots/grpc/proto/camera.proto,sha256=unLowpk4d0AGJgyKrbzTr1F_CxOV9dXRoXn-gO8eJVw,1097
+luckyrobots/grpc/proto/common.proto,sha256=zQPj-Z8b8mAE2_eHlJ0L2_-JH-CYY_OJcv8046fAtQM,772
+luckyrobots/grpc/proto/hazel_rpc.proto,sha256=gtFukd1B9WRbY2kWBgd5YQWSw4cobYf1YRYX6VEaULE,1229
+luckyrobots/grpc/proto/media.proto,sha256=3Rhmiaw1zceQBFdwxX1ViFgH-cSr7mBjNEGHPaFmTsk,836
+luckyrobots/grpc/proto/mujoco.proto,sha256=b7tBvYqqvcSjmqrEax6_75-P3fIHHZq63A1HVJKjx0U,1873
+luckyrobots/grpc/proto/scene.proto,sha256=y19i9MYkqZQPfrvTZEwMH_LekYGOIjvo5qtCXh9_Xlg,1825
+luckyrobots/grpc/proto/telemetry.proto,sha256=PZiUoA0bpwuvsBUdoQQdBNGWOuhlBMKAiBpwqPZiJgY,1481
+luckyrobots/grpc/proto/viewport.proto,sha256=gunF8cIrVgvoAX8FNER6gw_u-IlRWAR1Q6ts0zUuqr8,1324
+luckyrobots/models/__init__.py,sha256=assqJIKZXvDGwCVM0grNLJCBWyHtKRBAaxdJb2dON10,332
+luckyrobots/models/camera.py,sha256=e-CftAndHASfokhtUtU7ujGb2q_kcXZ6-2KKvY42CKg,3682
+luckyrobots/models/observation.py,sha256=jbRiUTjOP2ZiSobsnNcks3aMPAJA6oPoKqiXwvrNQK0,4758
+luckyrobots/models/randomization.py,sha256=AicqHovzG97Ge33wtNQE-_EvKxndjEHguTjCiHfhAV8,2753
+luckyrobots-0.1.71.dist-info/METADATA,sha256=9E-gglRuotJLzwUVxRy44pQlfDyCwKbImt_Zt7MwBT4,8379
+luckyrobots-0.1.71.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+luckyrobots-0.1.71.dist-info/licenses/LICENSE,sha256=xsPYvRJPH_fW_sofTUknI_KvZOsD4-BqjSOTZqI6Nmw,1069
+luckyrobots-0.1.71.dist-info/RECORD,,

luckyrobots-0.1.69.dist-info/RECORD

@@ -1,44 +0,0 @@
-luckyrobots/__init__.py,sha256=3hJT7tvGfzU6-PzXICpv4O9SNAiuFhE95o0_dTxLi7Y,739
-luckyrobots/client.py,sha256=D81pk16ZFKP_uYDSJAur_aucRu7eW2JXW0s63HAYWfQ,28935
-luckyrobots/luckyrobots.py,sha256=F68TUhKplLbdkuARGGUer9jSf1QUzLhu1Rfc-vqDZso,6788
-luckyrobots/utils.py,sha256=62BqpSRBFykVRMZ5Ti2F-yKRDoyU3UNHeesDF_PVCFc,2958
-luckyrobots/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-luckyrobots/config/robots.yaml,sha256=Wu2AgwhZxo9F-RLcqPJXHnqzjT8a0wR-rFBcecck_eE,5164
-luckyrobots/engine/__init__.py,sha256=-3r7aLueDF7eyumqIhTvYqS8ZNnVSJUJEHG4ybsKrjc,546
-luckyrobots/engine/check_updates.py,sha256=KGQHrj1LB0_vexktO01xx-ZvPOYKkto4VBb3H1KV7uI,7593
-luckyrobots/engine/download.py,sha256=vIb5T40Jk2BnfjPo0f4KgW14Om6P4_mvPpZ9JAlkVvg,4029
-luckyrobots/engine/manager.py,sha256=8vxSevaoXo-lw2G6y74lVc-Ms8xIELK0yLswydMiOQY,13388
-luckyrobots/grpc/__init__.py,sha256=hPh-X_FEus2sxrfQB2fQjD754X5bMaUFTWm3mZmUhe0,153
-luckyrobots/grpc/generated/__init__.py,sha256=u69M4E_2LwRQSD_n4wvQyMX7dryBCrIdsho59c52Crk,732
-luckyrobots/grpc/generated/agent_pb2.py,sha256=71ILNgwKLNhfiTsU8ocZTasG-hNViF03Cn3-ldBvc9A,5728
-luckyrobots/grpc/generated/agent_pb2_grpc.py,sha256=LAiYHz2UhQNFCKYmm4FeKnSIVXpAoR9CpWiIKTu1Bro,8962
-luckyrobots/grpc/generated/camera_pb2.py,sha256=QGxa8sfMqzzdShyA7RVXLWsfSzXZ0Kg7cPSZYYl6E2c,2664
-luckyrobots/grpc/generated/camera_pb2_grpc.py,sha256=u7jimo_r0ThMmQOl43IXoyWNGIX2T583mv4xcCBciqk,5212
-luckyrobots/grpc/generated/common_pb2.py,sha256=puH7AOneT8K5FjcW2PjnUY6e2u-vobtnS4bpaPl4nhM,2073
-luckyrobots/grpc/generated/common_pb2_grpc.py,sha256=kFXVJY9Ju1lO1ramSeiLiMAZBYWBDrDaON96j2C3rXI,901
-luckyrobots/grpc/generated/media_pb2.py,sha256=FUgow9NoFf9DztuVrrokGXxuJBHFb0AKBQAc3aez7BU,1839
-luckyrobots/grpc/generated/media_pb2_grpc.py,sha256=AqZmnqqFTTxtFC5bsA-YygCMD93BTRSb97RBovHCARo,900
-luckyrobots/grpc/generated/mujoco_pb2.py,sha256=20UVm-fptmjfv17L06HKSTkMyVZwKa-aPzaltAUaNSA,3540
-luckyrobots/grpc/generated/mujoco_pb2_grpc.py,sha256=HDroQ6_HgaG06WxLsMwo6DHze3Or8bGXlhJln8Oqwfo,8442
-luckyrobots/grpc/generated/scene_pb2.py,sha256=ggZ5ZqMoYyWY4MShDG6S7TTH7HIa3_uyddxFEkF4ocw,4067
-luckyrobots/grpc/generated/scene_pb2_grpc.py,sha256=yPWO10jLAL9orTtzG08NDjWVz1uw8i9Yf8zRVtESMHE,8417
-luckyrobots/grpc/generated/telemetry_pb2.py,sha256=RdTlqizfmLBlT3Es0A_gkW0ec4rY3AsMwXkuG8Ds8zU,2840
-luckyrobots/grpc/generated/telemetry_pb2_grpc.py,sha256=Sq6nExNUZe1-rERJceSoSt23Pbx-DeFA3zrOJsOQpxA,5397
-luckyrobots/grpc/generated/viewport_pb2.py,sha256=LO_6MujrgVjm_QgeGjsyBVHal13SXN_6ROGb86WAEso,3147
-luckyrobots/grpc/generated/viewport_pb2_grpc.py,sha256=QIrjDAdE2kU-R6VVzAso3iZDDdurJllXkXwBggFYCcs,5370
-luckyrobots/grpc/proto/agent.proto,sha256=-3MzIKGzDC6QhPYUHuCXZfp6ygXA7c9usWblV55np3s,5694
-luckyrobots/grpc/proto/camera.proto,sha256=fXCaU9TdJk0zHJs0Jbkq9DABOwaPJBHyUKj1RyMoJeQ,1100
-luckyrobots/grpc/proto/common.proto,sha256=6twE6_HQUMd1N6Ko2vUEDWtL07pCflHIfmit66ggvg0,775
-luckyrobots/grpc/proto/hazel_rpc.proto,sha256=a0F_Dl1JgmJdoIe4okL_tHDzIQvL7dETrb634EuWTrg,1232
-luckyrobots/grpc/proto/media.proto,sha256=CNZJTepdS6Gx68d_Cd3ciaPrOsAe3e7yZbgHIGYe1LY,839
-luckyrobots/grpc/proto/mujoco.proto,sha256=nVovXeedm-imERtSVLeBR8dpEDmlFxXhyVujKy9BylU,1876
-luckyrobots/grpc/proto/scene.proto,sha256=9p19X6_IdiarDXHczkxvjO6Uhjltbwnk8lYuZik7Q1s,1828
-luckyrobots/grpc/proto/telemetry.proto,sha256=vD27JPNPhgi36fkJMTLoFNbv5YXEiz1xntsK00OJ6-Q,1484
-luckyrobots/grpc/proto/viewport.proto,sha256=l3BZBho-v0DHLKt3OQhJwodaZMxZfFceAPQQUmW7PVU,1327
-luckyrobots/models/__init__.py,sha256=LsKj7q9fMTqpSC4WThJVlRMmwho5D08OURVCgerEURI,246
-luckyrobots/models/camera.py,sha256=e-CftAndHASfokhtUtU7ujGb2q_kcXZ6-2KKvY42CKg,3682
-luckyrobots/models/observation.py,sha256=jbRiUTjOP2ZiSobsnNcks3aMPAJA6oPoKqiXwvrNQK0,4758
-luckyrobots-0.1.69.dist-info/METADATA,sha256=S9czWJXY1XdNYfp2uPzs96YnxBqeJo9CD8ECBy1lQX0,8379
-luckyrobots-0.1.69.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-luckyrobots-0.1.69.dist-info/licenses/LICENSE,sha256=xsPYvRJPH_fW_sofTUknI_KvZOsD4-BqjSOTZqI6Nmw,1069
-luckyrobots-0.1.69.dist-info/RECORD,,