Fix the test matrix

comet-ml · Jul 25, 2024 · 9d349ff · 9d349ff
1 parent 479e696
commit 9d349ff
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/test-examples.yml b/.github/workflows/test-examples.yml
@@ -23,7 +23,6 @@ jobs:
           - integrations/model-optimization/ray-tune/notebooks/Comet_and_Ray.ipynb
           - integrations/model-training/composer/notebooks/comet_composer.ipynb
           - integrations/model-training/fastai/notebooks/fastai_hello_world.ipynb
-          - integrations/model-training/hugging_face/notebooks/Comet_with_Hugging_Face_Trainer.ipynb
           - integrations/model-training/keras/notebooks/Comet_with_Keras.ipynb
           - integrations/model-training/lightgbm/notebooks/Comet_and_LightGBM.ipynb
           - integrations/model-training/prophet/notebooks/Comet_and_Prophet.ipynb
@@ -33,9 +32,11 @@ jobs:
           - integrations/model-training/pytorch/notebooks/Comet_Pytorch_Tensorboard.ipynb
           - integrations/model-training/pytorch/notebooks/Comet_Pytorch_TensorboardX.ipynb
           - integrations/model-training/pytorch/notebooks/Histogram_Logging_Pytorch.ipynb
+          - integrations/model-training/ray-train/notebooks/Comet_with_ray_train_huggingface_transformers.ipynb
           - integrations/model-training/ray-train/notebooks/Comet_with_ray_train_keras.ipynb
           - integrations/model-training/ray-train/notebooks/Comet_with_ray_train_xgboost.ipynb
           - integrations/model-training/tensorflow/notebooks/Comet_and_Tensorflow.ipynb
+          - integrations/model-training/transformers/notebooks/Comet_with_Hugging_Face_Trainer.ipynb
           - integrations/model-training/yolov5/notebooks/Comet_and_YOLOv5.ipynb
           - integrations/model-training/yolov8/notebooks/YOLOv8_and_Comet.ipynb
           - integrations/reinforcement-learning/gymnasium/notebooks/comet_gymnasium_example.ipynb
@@ -107,7 +108,6 @@ jobs:
           - {script: "integrations/model-optimization/optuna/optuna-hello-world/optuna-hello-world.py", arg: ""}
           - {script: "integrations/model-training/composer/mosaicml-getting-started/mosaicml-getting-started.py", arg: ""}
           - {script: "integrations/model-training/fastai/fastai-hello-world/fastai_hello_world.py", arg: ""}
-          - {script: "integrations/model-training/hugging_face/transformers-distilbert-fine-tuning/transformers-distilbert-fine-tuning.py", arg: ""}
           - {script: "integrations/model-training/keras/keras-mnist-dnn/keras-mnist-dnn.py", arg: ""}
           - {script: "integrations/model-training/mlflow/mlflow-hello-world/mlflow-hello-world.py", arg: "run"}
           - {script: "integrations/model-training/pytorch-lightning/pytorch-lightning-optimizer/pytorch-lightning-optimizer.py", arg: ""}
@@ -116,6 +116,8 @@ jobs:
           - {script: "integrations/model-training/pytorch/pytorch-tensorboard/pytorch-tensorboard-example.py", arg: ""}
           - {script: "integrations/model-training/scikit-learn/sklearn-classification-example/comet-scikit-classification-example.py", arg: "run"}
           - {script: "integrations/model-training/scikit-learn/sklearn-nlp-example/comet-scikit-nlp-example.py", args: ""}
+          - {script: "integrations/model-training/transformers/transformers-distilbert-fine-tuning/transformers-distilbert-fine-tuning.py", arg: ""}
+          - {script: "integrations/model-training/transformers/transformers-google-bert-fine-tuning/transformers-google-bert-fine-tuning.py", arg: ""}
           - {script: "integrations/model-training/xgboost/xgboost-california/xgboost-california.py", arg: ""}
           - {script: "integrations/workflow-orchestration/metaflow/metaflow-hello-world/helloworld.py", arg: "run"}
           - {script: "integrations/workflow-orchestration/metaflow/metaflow-model-evaluation/metaflow-model-evaluation.py", arg: "run --max-workers 1 --n_samples 100"}