Spaces:
Sleeping
Sleeping
File size: 2,017 Bytes
be32845 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
# Copyright (c) Yogesh Singla and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
"""
Data models for the Julia Environment.
The Julia environment executes Julia code and provides feedback through
compilation and unit test results.
"""
from dataclasses import dataclass, field
from typing import Optional
from core.env_server.types import Action, Observation, State
@dataclass(kw_only=True)
class JuliaAction(Action):
"""
Action for the Julia environment - code to execute.
Attributes:
core_code: Core Julia code to execute
test_code: Test code to execute
"""
core_code: str
test_code: str
@dataclass(kw_only=True)
class JuliaObservation(Observation):
"""
Observation from the Julia environment - execution results.
Attributes:
stdout: Standard output from Julia execution
stderr: Standard error from Julia execution
exit_code: Exit code (0 = success, non-zero = error)
execution_time: Time taken to execute in seconds
tests_passed: Number of tests passed (if tests were run)
tests_failed: Number of tests failed (if tests were run)
code_compiles: Whether the core code compiled/executed successfully
"""
stdout: str = ""
stderr: str = ""
exit_code: int = 0
tests_passed: int = 0
tests_failed: int = 0
code_compiles: bool = True
@dataclass
class JuliaState(State):
"""
State for Julia environment.
Attributes:
episode_id: Unique episode identifier
step_count: Number of steps taken in episode
last_exit_code: Exit code from last execution
total_tests_passed: Cumulative tests passed in episode
total_tests_failed: Cumulative tests failed in episode
"""
last_exit_code: int = 0
last_code_compiles: bool = True
total_tests_passed: int = 0
total_tests_failed: int = 0
|