put in agent parsing skeleton

This commit is contained in:
Marek Wolan
2023-09-26 12:54:56 +01:00
parent 92e0110e73
commit f1346ae278
11 changed files with 86 additions and 54 deletions

View File

@@ -0,0 +1,20 @@
from abc import ABC, abstractmethod
from typing import Any, Dict, List
from pydantic import BaseModel
class AbstractReward(BaseModel):
def __call__(self, states: List[Dict]) -> float:
"""_summary_
:param state: _description_
:type state: Dict
:return: _description_
:rtype: float
"""
...
class RewardFunction(BaseModel):
...