Difficulties implementing my own plugin metric

[ReemSharwalla]

I created some metrics that evaluate ML-Models in this format:

class Metric:
def init(self, ...., curve=False):
...
self.curve = curve

def update(self, outputs, labels):    
    self.list1.extend() 
    self.list2.extend() 

def compute(self):
    .......  # some computations
    # Plot the curve if enabled
    if self.curve:
        self.plot_curve(.....)
    return result

def plot_curve(self, .....):
    ........
    # Show the plot
    plt.show()

But i have difficulties implementing it in the CL model. Can please someone help me with this? Should I change the format?