graders/medium_grader.py

from graders.base import BaseGrader


class MediumGrader(BaseGrader):
    def grade(self, trajectory, ground_truth):
        if len(trajectory) > 1:
            return 0.6
        return 0.2