def predict()

in cluster-trace-gpu-v2020/simulator/job_history.py [0:0]

• 10 lines of code
• 4 McCabe index (conditional complexity)

    def predict(self, user, metrics=None):
        """
        metrics: in ['dur_avg', 'dur_har', 'dur_mva']
        """

        metrics = 'dur_avg' if metrics is None else metrics
        assert metrics in ['dur_avg', 'dur_har', 'dur_mva']
        if user not in self.user_job_stats:
            return 0  # No record => optimistic est. as 0.
        else:
            res = self.user_job_stats[user][metrics]
            if np.isnan(res):
                raise TypeError
            return res