|
808 | 808 | " )\n",
|
809 | 809 | ")"
|
810 | 810 | ]
|
811 |
| - }, |
812 |
| - { |
813 |
| - "cell_type": "markdown", |
814 |
| - "metadata": {}, |
815 |
| - "source": [ |
816 |
| - "## Clean up\n", |
817 |
| - "Delete the model registry and the pipeline after you complete the lab." |
818 |
| - ] |
819 |
| - }, |
820 |
| - { |
821 |
| - "cell_type": "code", |
822 |
| - "execution_count": null, |
823 |
| - "metadata": {}, |
824 |
| - "outputs": [], |
825 |
| - "source": [ |
826 |
| - "def delete_model_package_group(sm_client, package_group_name):\n", |
827 |
| - " try:\n", |
828 |
| - " model_versions = sm_client.list_model_packages(ModelPackageGroupName=package_group_name)\n", |
829 |
| - "\n", |
830 |
| - " except Exception as e:\n", |
831 |
| - " print(\"{} \\n\".format(e))\n", |
832 |
| - " return\n", |
833 |
| - "\n", |
834 |
| - " for model_version in model_versions[\"ModelPackageSummaryList\"]:\n", |
835 |
| - " try:\n", |
836 |
| - " sm_client.delete_model_package(ModelPackageName=model_version[\"ModelPackageArn\"])\n", |
837 |
| - " except Exception as e:\n", |
838 |
| - " print(\"{} \\n\".format(e))\n", |
839 |
| - " time.sleep(0.5) # Ensure requests aren't throttled\n", |
840 |
| - "\n", |
841 |
| - " try:\n", |
842 |
| - " sm_client.delete_model_package_group(ModelPackageGroupName=package_group_name)\n", |
843 |
| - " print(\"{} model package group deleted\".format(package_group_name))\n", |
844 |
| - " except Exception as e:\n", |
845 |
| - " print(\"{} \\n\".format(e))\n", |
846 |
| - " return\n", |
847 |
| - "\n", |
848 |
| - "\n", |
849 |
| - "def delete_sagemaker_pipeline(sm_client, pipeline_name):\n", |
850 |
| - " try:\n", |
851 |
| - " sm_client.delete_pipeline(\n", |
852 |
| - " PipelineName=pipeline_name,\n", |
853 |
| - " )\n", |
854 |
| - " print(\"{} pipeline deleted\".format(pipeline_name))\n", |
855 |
| - " except Exception as e:\n", |
856 |
| - " print(\"{} \\n\".format(e))\n", |
857 |
| - " return\n", |
858 |
| - " \n", |
859 |
| - "def delete_sagemaker_project(sm_client, project_name):\n", |
860 |
| - " try:\n", |
861 |
| - " sm_client.delete_project(\n", |
862 |
| - " ProjectName=project_name,\n", |
863 |
| - " )\n", |
864 |
| - " print(\"{} project deleted\".format(project_name))\n", |
865 |
| - " except Exception as e:\n", |
866 |
| - " print(\"{} \\n\".format(e))\n", |
867 |
| - " return" |
868 |
| - ] |
869 |
| - }, |
870 |
| - { |
871 |
| - "cell_type": "code", |
872 |
| - "execution_count": null, |
873 |
| - "metadata": {}, |
874 |
| - "outputs": [], |
875 |
| - "source": [ |
876 |
| - "import boto3\n", |
877 |
| - "import time\n", |
878 |
| - "\n", |
879 |
| - "client = boto3.client(\"sagemaker\")\n", |
880 |
| - "\n", |
881 |
| - "# Uncomment the lines below to clean the pipeline.\n", |
882 |
| - "#delete_model_package_group(client, model_package_group_name)\n", |
883 |
| - "#delete_sagemaker_pipeline(client, pipeline_name)\n", |
884 |
| - "\n", |
885 |
| - "#delete_model_package_group(client, model_package_group_name2)\n", |
886 |
| - "#delete_sagemaker_pipeline(client, pipeline_name2)\n", |
887 |
| - "\n", |
888 |
| - "# delete_sagemaker_project(client, \"<Your-Project-Name>\")#\"cv-week4-training\") #" |
889 |
| - ] |
890 | 811 | }
|
891 | 812 | ],
|
892 | 813 | "metadata": {
|
|
0 commit comments