def after_fit(self):
    if (not self.keep_experiment_running):
        try:
            self.experiment.stop()
        except:
            print('No neptune experiment to stop.')
    else:
        print(f'''Your experiment (id: {self.experiment.id}, name: {self.experiment.name}) is left in the running state.
''', 'You can log more data to it, like this: `neptune.log_metric()`')
def after_fit(self):
    if (not self.keep_experiment_running):
        try:
            self.experiment.stop()
        except:
            print('No neptune experiment to stop.')
    else:
        print(f'Your experiment (id: {self.experiment.id}, name: {self.experiment.name}) is left in the running state.
', 'You can log more data to it, like this: `neptune.log_metric()`')