chore: LLM - Changed the model distillation pipeline URI

vertex-sdk-bot · copybara-github · commit 6e6d00524f69 · 2023-12-20T18:31:06.000-08:00
PiperOrigin-RevId: 592708955
diff --git a/tests/unit/aiplatform/test_language_models.py b/tests/unit/aiplatform/test_language_models.py
@@ -994,7 +994,7 @@ def reverse_string_2(s):""",
 )
 
 _URL_DATA = {
-    "https://us-kfp.pkg.dev/ml-pipeline/research/distillation/v1.0.0": _TEST_DISTILLATION_PIPELINE_SPEC_JSON,
+    "https://us-kfp.pkg.dev/ml-pipeline/distillation/distillation/v1.0.0": _TEST_DISTILLATION_PIPELINE_SPEC_JSON,
 }
 
 
@@ -4395,7 +4395,7 @@ def test_model_evaluation_text_classification_base_model_only_summary_metrics(
     )
     @pytest.mark.parametrize(
         "mock_urllib_request_urlopen",
-        ["https://us-kfp.pkg.dev/ml-pipeline/research/distillation/v1.0.0"],
+        ["https://us-kfp.pkg.dev/ml-pipeline/distillation/distillation/v1.0.0"],
         indirect=True,
     )
     def test_text_generation_model_distill_from(
diff --git a/vertexai/language_models/_distillation.py b/vertexai/language_models/_distillation.py
@@ -8,7 +8,7 @@
 
 class DistillationMixin:
     _DISTILLATION_PIPELINE_URI = (
-        "https://us-kfp.pkg.dev/ml-pipeline/research/distillation/v1.0.0"
+        "https://us-kfp.pkg.dev/ml-pipeline/distillation/distillation/v1.0.0"
     )
 
     def distill_from(

Original file line number	Diff line number	Diff line change
`@@ -994,7 +994,7 @@ def reverse_string_2(s):""",`
`994`	`994`	`)`
`995`	`995`
`996`	`996`	`_URL_DATA = {`
`997`		`- "https://us-kfp.pkg.dev/ml-pipeline/research/distillation/v1.0.0": _TEST_DISTILLATION_PIPELINE_SPEC_JSON,`
	`997`	`+ "https://us-kfp.pkg.dev/ml-pipeline/distillation/distillation/v1.0.0": _TEST_DISTILLATION_PIPELINE_SPEC_JSON,`
`998`	`998`	`}`
`999`	`999`
`1000`	`1000`
`@@ -4395,7 +4395,7 @@ def test_model_evaluation_text_classification_base_model_only_summary_metrics(`
`4395`	`4395`	`)`
`4396`	`4396`	`@pytest.mark.parametrize(`
`4397`	`4397`	`"mock_urllib_request_urlopen",`
`4398`		`- ["https://us-kfp.pkg.dev/ml-pipeline/research/distillation/v1.0.0"],`
	`4398`	`+ ["https://us-kfp.pkg.dev/ml-pipeline/distillation/distillation/v1.0.0"],`
`4399`	`4399`	`indirect=True,`
`4400`	`4400`	`)`
`4401`	`4401`	`def test_text_generation_model_distill_from(`
Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`
`9`	`9`	`class DistillationMixin:`
`10`	`10`	`_DISTILLATION_PIPELINE_URI = (`
`11`		`- "https://us-kfp.pkg.dev/ml-pipeline/research/distillation/v1.0.0"`
	`11`	`+ "https://us-kfp.pkg.dev/ml-pipeline/distillation/distillation/v1.0.0"`
`12`	`12`	`)`
`13`	`13`
`14`	`14`	`def distill_from(`