from repro.models.yuan2021 import BARTScore, DEFAULT_IMAGE

from sacrerouge.metrics import Metric
from sacrerouge.metrics.docker import ReferenceBasedDockerMetric


@Metric.register('docker-bartscore')
class DockerBARTScore(ReferenceBasedDockerMetric):
    def __init__(
        self,
        image: str = DEFAULT_IMAGE,
        model: str = "default",
        device: int = 0,
    ):
        metric = BARTScore(
            image=image,
            model=model,
            device=device,
        )
        super().__init__(metric)

ba3304658b4f5e20478a0ffe291c8246494ab7a9
<mask token>

<mask token>


@Metric.register('docker-bartscore')
class DockerBARTScore(ReferenceBasedDockerMetric):
    <mask token>

<mask token>


@Metric.register('docker-bartscore')
class DockerBARTScore(ReferenceBasedDockerMetric):

    def __init__(self, image: str=DEFAULT_IMAGE, model: str='default',
        device: int=0):
        metric = BARTScore(image=image, model=model, device=device)
        super().__init__(metric)

from repro.models.yuan2021 import BARTScore, DEFAULT_IMAGE
from sacrerouge.metrics import Metric
from sacrerouge.metrics.docker import ReferenceBasedDockerMetric


@Metric.register('docker-bartscore')
class DockerBARTScore(ReferenceBasedDockerMetric):

    def __init__(self, image: str=DEFAULT_IMAGE, model: str='default',
        device: int=0):
        metric = BARTScore(image=image, model=model, device=device)
        super().__init__(metric)

from repro.models.yuan2021 import BARTScore, DEFAULT_IMAGE

from sacrerouge.metrics import Metric
from sacrerouge.metrics.docker import ReferenceBasedDockerMetric


@Metric.register('docker-bartscore')
class DockerBARTScore(ReferenceBasedDockerMetric):
    def __init__(
        self,
        image: str = DEFAULT_IMAGE,
        model: str = "default",
        device: int = 0,
    ):
        metric = BARTScore(
            image=image,
            model=model,
            device=device,
        )
        super().__init__(metric)
