Spaces:

TheCrustaceans
/

Pm-ops

Sleeping

Pm-ops / client.py

made minor type check changes, open env has strict type checking

74d0de6 7 months ago

469 Bytes

"""OpenEnv client wrapper for PM-Ops environment."""
import os
from openenv.core.env_client import EnvClient
from models import PMOpsAction, PMOpsObservation


class PMOpsEnv(EnvClient):
    action_class = PMOpsAction
    observation_class = PMOpsObservation

    def __init__(self, base_url: str | None = None, token: str | None= None):
        url = base_url or os.getenv("API_BASE_URL", "https://adityaguntur-pm-ops.hf.space")
        super().__init__(base_url=url)

	"""OpenEnv client wrapper for PM-Ops environment."""
	import os
	from openenv.core.env_client import EnvClient
	from models import PMOpsAction, PMOpsObservation


	class PMOpsEnv(EnvClient):
	action_class = PMOpsAction
	observation_class = PMOpsObservation

	def __init__(self, base_url: str \| None = None, token: str \| None= None):
	url = base_url or os.getenv("API_BASE_URL", "https://adityaguntur-pm-ops.hf.space")
	super().__init__(base_url=url)