Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 0 additions & 18 deletions predicators/agent_sdk/belief_probe.py
Original file line number Diff line number Diff line change
Expand Up @@ -1205,20 +1205,6 @@ def predicates(self,
max_trajectories=max_trajectories,
max_groundings_per_predicate=max_groundings_per_predicate)

def samplers(self) -> str:
"""Reload ``samplers.py`` and install its per-skill samplers.

Sampler-synthesis sessions only. Loads ``LEARNED_SAMPLERS``
fresh from the file (snapshotting it into
``samplers_versions/``), validates the option-name -> callable
map, installs it so ``refine`` draws from the draft samplers,
and reports a per-option sanity check (return shape, in-box
draws) on a representative train-task state. Call it after every
edit of ``samplers.py``.
"""
self._require_available("samplers")
return self._artifact_loader("samplers")()

def _artifact_loader(self, name: str) -> Callable[..., str]:
ctx = self._ctx
_check_time_budget(ctx)
Expand Down Expand Up @@ -2903,7 +2889,6 @@ def _on_rollout() -> None:
rng=rng,
max_draws=max(1, int(max_draws)),
top_k=max(1, int(top_k)),
parameterized_samplers=ctx.parameterized_samplers or None,
on_rollout=_on_rollout)
return ProbeSuggestResult(suggestions, list(notices) + notes)

Expand Down Expand Up @@ -2974,7 +2959,6 @@ def plan_scorer(plan: List[Any],
rng=rng,
max_draws=max(1, max_draws),
top_k=max(1, int(top_k)),
parameterized_samplers=ctx.parameterized_samplers or None,
on_rollout=lambda: _check_time_budget(ctx),
plan_scorer=plan_scorer)
notices.append(
Expand Down Expand Up @@ -3095,7 +3079,6 @@ def gated_solved_check(states: List[State], labels: List[Any],
check_subgoals=True,
check_final_goal=require_goal,
run_id="probe",
parameterized_samplers=ctx.parameterized_samplers or None,
strip_latent_wait_targets=not ctx.latent_tracking_available,
solved_check=solved_check)
refined_plan, success = outcome.plan, outcome.success
Expand Down Expand Up @@ -3219,7 +3202,6 @@ def _select_on_joint_draws(
check_subgoals=True,
check_final_goal=require_goal,
run_id="probe",
parameterized_samplers=ctx.parameterized_samplers or None,
strip_latent_wait_targets=not ctx.latent_tracking_available,
solved_check=solved_check)
extra_samples += outcome.total_samples
Expand Down
4 changes: 0 additions & 4 deletions predicators/agent_sdk/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,7 @@ class SessionConfig:
max_turns: int
max_buffer_size: int
agent_timeout: int
use_docker_sandbox: bool
use_local_sandbox: bool
docker_image: str
use_scratchpad: bool

@classmethod
Expand All @@ -44,9 +42,7 @@ def from_cfg(cls) -> "SessionConfig":
max_turns=CFG.agent_sdk_max_agent_turns_per_iteration,
max_buffer_size=CFG.agent_sdk_max_buffer_size,
agent_timeout=CFG.agent_sdk_agent_timeout,
use_docker_sandbox=CFG.agent_sdk_use_docker_sandbox,
use_local_sandbox=CFG.agent_sdk_use_local_sandbox,
docker_image=CFG.agent_sdk_docker_image,
use_scratchpad=CFG.agent_planner_use_scratchpad,
)

Expand Down
Loading
Loading