Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
LOCATION = "us-central1"
MODEL_ID = "3125638878883479552" # bq all
MODEL_ID = "8842430840248991744" # bq all
DISPLAY_NAME = f"temp_create_batch_prediction_job_test_{uuid4()}"
BIGQUERY_SOURCE_INPUT_URI = "bq://ucaip-sample-tests.table_test.all_bq_types"
BIGQUERY_DESTINATION_OUTPUT_URI = "bq://ucaip-sample-tests"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
LOCATION = "us-central1"
MODEL_ID = "1478306577684365312" # Permanent 50 flowers model
MODEL_ID = "3512561418744365056" # Permanent 50 flowers model
DISPLAY_NAME = f"temp_create_batch_prediction_job_test_{uuid4()}"
GCS_SOURCE_URI = (
"gs://ucaip-samples-test-output/inputs/icn_batch_prediction_input.jsonl"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
LOCATION = "us-central1"
MODEL_ID = "3863595899074641920" # Permanent restaurant rating model
MODEL_ID = "7827432074230366208" # Permanent restaurant rating model
DISPLAY_NAME = f"temp_create_batch_prediction_tcn_test_{uuid4()}"
GCS_SOURCE_URI = (
"gs://ucaip-samples-test-output/inputs/batch_predict_TCN/tcn_inputs.jsonl"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
LOCATION = "us-central1"
MODEL_ID = "5216364637146054656" # Permanent medical entity NL model
MODEL_ID = "6305215400179138560" # Permanent medical entity NL model
DISPLAY_NAME = f"temp_create_batch_prediction_ten_test_{uuid4()}"
GCS_SOURCE_URI = (
"gs://ucaip-samples-test-output/inputs/batch_predict_TEN/ten_inputs.jsonl"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
LOCATION = "us-central1"
MODEL_ID = "667940119734386688" # Permanent 5 class sports model
MODEL_ID = "8596984660557299712" # Permanent 5 class sports model
DISPLAY_NAME = f"temp_create_batch_prediction_vcn_test_{uuid4()}"
GCS_SOURCE_URI = (
"gs://ucaip-samples-test-output/inputs/vcn_40_batch_prediction_input.jsonl"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
LOCATION = "us-central1"
MODEL_ID = "20547673299877888" # Permanent horses model
MODEL_ID = "8609932509485989888" # Permanent horses model
DISPLAY_NAME = f"temp_create_batch_prediction_vot_test_{uuid4()}"
GCS_SOURCE_URI = (
"gs://ucaip-samples-test-output/inputs/vot_batch_prediction_input.jsonl"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import export_model_tabular_classification_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "5359002081594179584" # iris 1000
MODEL_ID = "6036688272397172736" # iris 1000
GCS_BUCKET = "gs://ucaip-samples-test-output"
GCS_PREFIX = f"tmp/export_model_test_{uuid4()}"

Expand Down
4 changes: 2 additions & 2 deletions samples/snippets/get_model_evaluation_sample_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@
import get_model_evaluation_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "5162251072873431040" # permanent_safe_driver_model
EVALUATION_ID = "5615675837586029221" # permanent_safe_driver_model Evaluation
MODEL_ID = "3512561418744365056" # permanent_safe_driver_model
EVALUATION_ID = "9035588644970168320" # permanent_safe_driver_model Evaluation


def test_ucaip_generated_get_model_evaluation_sample(capsys):
Expand Down
6 changes: 3 additions & 3 deletions samples/snippets/get_model_evaluation_slice_sample_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,9 @@
import get_model_evaluation_slice_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "5162251072873431040" # permanent_safe_driver_model
EVALUATION_ID = "5615675837586029221" # permanent_safe_driver_model Evaluation
SLICE_ID = "4322488217836113260" # permanent_safe_driver_model Eval Slice
MODEL_ID = "3512561418744365056" # permanent_safe_driver_model
EVALUATION_ID = "9035588644970168320" # permanent_safe_driver_model Evaluation
SLICE_ID = "6481571820677004173" # permanent_safe_driver_model Eval Slice


def test_ucaip_generated_get_model_evaluation_slice_sample(capsys):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@
import get_model_evaluation_tabular_classification_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "5162251072873431040" # permanent_safe_driver_model
EVALUATION_ID = "5615675837586029221" # permanent_safe_driver_model Evaluation
MODEL_ID = "6036688272397172736" # permanent_safe_driver_model
EVALUATION_ID = "1866113044163962838" # permanent_safe_driver_model Evaluation


def test_ucaip_generated_get_model_evaluation_tabular_classification_sample(capsys):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@
import get_model_evaluation_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "3125638878883479552" # bq all
EVALUATION_ID = "2025948722346981108" # bq all evaluation
MODEL_ID = "8842430840248991744" # bq all
EVALUATION_ID = "4944816689650806017" # bq all evaluation


def test_ucaip_generated_get_model_evaluation_sample(capsys):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@
import get_model_evaluation_video_classification_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "667940119734386688" # Permanent sports model
EVALUATION_ID = "789396572185034752" # Permanent sports evaluation
MODEL_ID = "8596984660557299712" # Permanent sports model
EVALUATION_ID = "7092045712224944128" # Permanent sports evaluation


def test_ucaip_generated_get_model_evaluation_sample(capsys):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@
import get_model_evaluation_video_object_tracking_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "20547673299877888" # Permanent horses model
EVALUATION_ID = "1165447141070471168" # Permanent horses evaluation
MODEL_ID = "8609932509485989888" # Permanent horses model
EVALUATION_ID = "6016811301190238208" # Permanent horses evaluation


def test_ucaip_generated_get_model_evaluation_sample(capsys):
Expand Down
2 changes: 1 addition & 1 deletion samples/snippets/get_model_sample_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
import get_model_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "1478306577684365312" # permanent_50_flowers_model
MODEL_ID = "3512561418744365056" # permanent_50_flowers_model
KNOWN_MODEL = f"/locations/us-central1/models/{MODEL_ID}"


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ def dataset_name():
parent=f"projects/{PROJECT_ID}/locations/{LOCATION}", dataset=dataset
)

created_dataset = operation.result(timeout=120)
created_dataset = operation.result(timeout=600)

yield created_dataset.name

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ def test_import_data_video_action_recognition_sample(
parent=f"projects/{PROJECT_ID}/locations/{LOCATION}", dataset=dataset
)

create_dataset_response = response.result(timeout=120)
create_dataset_response = response.result(timeout=600)

shared_state["dataset_name"] = create_dataset_response.name
shared_state["dataset_id"] = create_dataset_response.name.split("/")[-1]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,10 @@
import list_model_evaluation_slices_sample

PROJECT_ID = os.getenv("BUILD_SPECIFIC_GCLOUD_PROJECT")
MODEL_ID = "5162251072873431040"
EVALUATION_ID = "5615675837586029221"
MODEL_ID = "3512561418744365056"
EVALUATION_ID = "9035588644970168320"

KNOWN_EVALUATION_SLICE = "/locations/us-central1/models/5162251072873431040/evaluations/5615675837586029221/slices/4322488217836113260"
KNOWN_EVALUATION_SLICE = "projects/580378083368/locations/us-central1/models/3512561418744365056/evaluations/9035588644970168320/slices/6481571820677004173"


def test_ucaip_generated_get_model_evaluation_slices_sample(capsys):
Expand Down