Esempio n. 1
0
def test_callback_is_callable():
    cb = __MLflowGluonCallback(
        log_models=True, metrics_logger=BatchMetricsLogger(run_id="1234"))
    pickle.dumps(cb)
Esempio n. 2
0
 def getGluonCallback(metrics_logger):
     return __MLflowGluonCallback(log_models, metrics_logger)