--- a +++ b/aiagents4pharma/talk2biomodels/tools/utils.py @@ -0,0 +1,22 @@ +#!/usr/bin/env python3 + +""" +Utility functions for T2B tools. +""" + +import basico + +def get_model_units(model_object): + """ + Get the units of the model. + + Args: + model_object: The model object. + + Returns: + dict: The units of the model. + """ + model_units = basico.model_info.get_model_units(model=model_object.copasi_model) + model_units_y = model_units['quantity_unit'] + model_units_x = model_units['time_unit'] + return {'y_axis_label': model_units_y, 'x_axis_label': model_units_x}