def get_manual_weights()

in datasets/MixedPrioritizedSampler.py [0:0]

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

    def get_manual_weights(self, lam):
        # Merge balanced weights and uniform weights 
        if lam == 1:
            manual_weights = self.balanced_weights
        elif lam == 0:
            manual_weights = np.ones(len(self.balanced_weights))
        else:
            manual_weights = self.balanced_weights * lam + (1-lam)
        return manual_weights