remove comment

Author: Laure-di

docs/resources/inference_custom_model.md

@@ -51,6 +51,8 @@ resource "scaleway_inference_deployment" "main" {
 
 ## Attributes Reference
 
+In addition to all arguments above, the following attributes are exported:
+
 - `id` - The unique identifier of the custom model.
 - `tags` - Tags associated with the model.
 - `status` - The current status of the model (e.g., ready, error, etc.).

internal/services/inference/custom_model.go

@@ -193,8 +193,8 @@ func ResourceCustomModelRead(ctx context.Context, d *schema.ResourceData, m inte
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("parameter_size_bits", model.ParameterSizeBits)
-	_ = d.Set("size_bits", model.SizeBytes)
+	_ = d.Set("parameter_size_bits", int32(model.ParameterSizeBits))
+	_ = d.Set("size_bits", int64(model.SizeBytes))
 	_ = d.Set("name", model.Name)
 	_ = d.Set("status", model.Status)
 	_ = d.Set("description", model.Description)

internal/services/inference/waiter.go

@@ -31,7 +31,7 @@ func waitForModel(ctx context.Context, inferenceAPI *inference.API, region scw.R
 		retryInterval = *transport.DefaultWaitRetryInterval
 	}
 
-	model, err := inferenceAPI.WaitForCustomModel(inference.WaitForCustomModelRequest{ //TODO add pointer
+	model, err := inferenceAPI.WaitForCustomModel(inference.WaitForCustomModelRequest{
 		ModelID:       id,
 		Region:        region,
 		RetryInterval: &retryInterval,