Fixing finetune checker test and doc (#150)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

Author: thiago-aixplain

docs/user/user_doc.md

@@ -352,9 +352,12 @@ Once a `FineTune` is created (refer to the [section above](#creating-a-finetune)
 ```python
 finetune_model = finetune.start()
 ```
-We receive a model that we can check the fine-tuning status:
+We receive a model that we can check the fine-tuning info (status, epoch, training and validation losses):
  ```python
-status = finetune_model.check_finetune_status()
+finetune_model_info = finetune_model.check_finetune_status()
+epoch = finetune_model_info.epoch
+training_loss = finetune_model_info.training_loss
+validation_loss = finetune_model_info.validation_loss
 ```
 Status can be one of the following: `onboarding`, `onboarded`, `hidden`, `training`, `deleted`, `enabling`, `disabled`, `failed`, `deleting`.
 

tests/functional/finetune/finetune_functional_test.py

@@ -76,13 +76,13 @@ def test_end2end(run_input_map):
     assert "inferenceCost" in cost_map
     finetune_model = finetune.start()
     start, end = time.time(), time.time()
-    status = finetune_model.check_finetune_status()
+    status = finetune_model.check_finetune_status().model_status.value
     while status != "onboarded" and (end - start) < TIMEOUT:
-        status = finetune_model.check_finetune_status()
+        status = finetune_model.check_finetune_status().model_status.value
         assert status != "failed"
         time.sleep(5)
         end = time.time()
-    assert finetune_model.check_finetune_status() == "onboarded"
+    assert finetune_model.check_finetune_status().model_status.value == "onboarded"
     result = finetune_model.run(run_input_map["inference_data"])
     print(f"Result: {result}")
     assert result is not None
@@ -128,4 +128,4 @@ def test_prompt_validator(validate_prompt_input_map):
             finetune = FinetuneFactory.create(
                 str(uuid.uuid4()), dataset_list, model, prompt_template=validate_prompt_input_map["prompt_template"]
             )
-        assert exc_info.type is AssertionError
+        assert exc_info.type is AssertionError