c: Update to use nemo runners (#728)

* Minor change to test CI

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Remove example test

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Minor change

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Run on NeMo GPU runners

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

---------

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Author: chtruong814

.github/workflows/gpuci.yml

@@ -76,7 +76,7 @@ jobs:
   run-gpu-tests:
     needs: [build-container]
     if: ${{ needs.changed-files.outputs.any_changed == 'true' }}
-    runs-on: linux-amd64-gpu-rtxa6000-latest-1
+    runs-on: linux-amd64-gpu-rtxa6000-latest-2-nemo
     environment: nemo-ci
     # Unit tests should not take longer than 40 minutes including docker pull and startup time
     timeout-minutes: 40

nemo_curator/hello.py

@@ -26,6 +26,7 @@ def create_logger(
     log_level: int = logging.INFO,
     stdout: bool = False,
 ) -> logging.Logger:
+    # Minor change
     # Create the logger
     logger = logging.getLogger(name)
     logger.setLevel(log_level)