benchmax 0.1.2.dev20__tar.gz → 0.1.2.dev22__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (52) hide show
  1. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/PKG-INFO +1 -1
  2. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/pyproject.toml +1 -1
  3. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/base_env.py +9 -0
  4. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax.egg-info/PKG-INFO +1 -1
  5. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/LICENSE +0 -0
  6. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/README.md +0 -0
  7. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/setup.cfg +0 -0
  8. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/adapters/__init__.py +0 -0
  9. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/adapters/benchmax_wrapper.py +0 -0
  10. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/adapters/skyrl/benchmax_data_process.py +0 -0
  11. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/adapters/skyrl/skyrl_adapter.py +0 -0
  12. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/bundle/__init__.py +0 -0
  13. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/bundle/bundler.py +0 -0
  14. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/bundle/errors.py +0 -0
  15. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/bundle/loader.py +0 -0
  16. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/bundle/payload.py +0 -0
  17. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/bundle/validator.py +0 -0
  18. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/__init__.py +0 -0
  19. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/crm/crm_env.py +0 -0
  20. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/crm/workdir/reward_fn.py +0 -0
  21. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/crm/workdir/salesforce_mcp.py +0 -0
  22. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/excel/data_utils.py +0 -0
  23. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/excel/excel_env.py +0 -0
  24. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/excel/workdir/__init__.py +0 -0
  25. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/excel/workdir/excel_code_runner_mcp.py +0 -0
  26. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/excel/workdir/excel_utils.py +0 -0
  27. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/excel/workdir/reward_fn.py +0 -0
  28. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/math/math_env.py +0 -0
  29. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/math/workdir/reward_fn.py +0 -0
  30. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/__init__.py +0 -0
  31. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/example_workdir/demo_mcp_server.py +0 -0
  32. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/example_workdir/reward_fn.py +0 -0
  33. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/parallel_mcp_env.py +0 -0
  34. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/provisioners/__init__.py +0 -0
  35. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/provisioners/base_provisioner.py +0 -0
  36. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/provisioners/local_provisioner.py +0 -0
  37. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/provisioners/manual_provisioner.py +0 -0
  38. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/provisioners/skypilot_provisioner.py +0 -0
  39. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/provisioners/utils.py +0 -0
  40. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/proxy_server.py +0 -0
  41. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/server_pool.py +0 -0
  42. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/mcp/utils.py +0 -0
  43. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/tracking.py +0 -0
  44. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/types.py +0 -0
  45. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/wikipedia/utils.py +0 -0
  46. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/wikipedia/wiki_env.py +0 -0
  47. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/prompts/__init__.py +0 -0
  48. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/prompts/tools.py +0 -0
  49. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax.egg-info/SOURCES.txt +0 -0
  50. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax.egg-info/dependency_links.txt +0 -0
  51. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax.egg-info/requires.txt +0 -0
  52. {benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax.egg-info/top_level.txt +0 -0
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: benchmax
3
- Version: 0.1.2.dev20
3
+ Version: 0.1.2.dev22
4
4
  Summary: Framework-Agnostic RL Environments for LLM Fine-Tuning
5
5
  Author: cgft.io
6
6
  Classifier: Programming Language :: Python :: 3
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "benchmax"
3
- version = "0.1.2.dev20"
3
+ version = "0.1.2.dev22"
4
4
  description = "Framework-Agnostic RL Environments for LLM Fine-Tuning"
5
5
  readme = "README.md"
6
6
  authors = [{ name = "cgft.io" }]
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax/envs/base_env.py RENAMED
@@ -38,6 +38,15 @@ class BaseEnv(ABC):
38
38
 
39
39
  self.compute_reward = _tracked
40
40
 
41
+ cls_compute_group_reward = type(self).compute_group_reward
42
+
43
+ @wraps(cls_compute_group_reward)
44
+ async def _tracked_group(*args, **kwargs):
45
+ with tracking_context(self._tracking_config):
46
+ return await cls_compute_group_reward(self, *args, **kwargs)
47
+
48
+ self.compute_group_reward = _tracked_group
49
+
41
50
  def get_tracking_config(self) -> Optional[TrackingConfig]:
42
51
  return self._tracking_config
43
52
 
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/src/benchmax.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: benchmax
3
- Version: 0.1.2.dev20
3
+ Version: 0.1.2.dev22
4
4
  Summary: Framework-Agnostic RL Environments for LLM Fine-Tuning
5
5
  Author: cgft.io
6
6
  Classifier: Programming Language :: Python :: 3
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/LICENSE RENAMED
File without changes
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/README.md RENAMED
File without changes
{benchmax-0.1.2.dev20 → benchmax-0.1.2.dev22}/setup.cfg RENAMED
File without changes