mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-10 03:30:58 +00:00
registry refactor
This commit is contained in:
parent
c50686b6fe
commit
95fd53d292
8 changed files with 39 additions and 71 deletions
|
|
@ -217,18 +217,8 @@ class BaseScorer(ABC, Generic[TScorerInputSample]):
|
|||
|
||||
|
||||
class BaseTask(ABC):
|
||||
def __init__(
|
||||
self,
|
||||
generator_processor: Optional[BaseGeneratorProcessor] = None,
|
||||
generator: Optional[BaseGenerator] = None,
|
||||
scorer: Optional[BaseScorer] = None,
|
||||
*args,
|
||||
**kwargs
|
||||
) -> None:
|
||||
def __init__(self, *args, **kwargs) -> None:
|
||||
super().__init__(*args, **kwargs)
|
||||
self.generator_processor = generator_processor
|
||||
self.generator = generator
|
||||
self.scorer = scorer
|
||||
|
||||
@abstractmethod
|
||||
async def run(self, *args, **kwargs) -> EvalResult:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue