Merge pull request #181 from GATEOverflow/dev

Added support for mlcd, mlce and mlcp shortcuts

Author: arjunsuresh

.github/workflows/test-mlc-core-actions.yaml

@@ -206,7 +206,7 @@ jobs:
     - name: Test 22 - Test recursive mlc pull repo
       run: |
         export MLC_REPOS=$HOME/test
-        mlc pull repo https://github.com/GATEOverflow/GO-PDFs
+        mlcp https://github.com/GATEOverflow/GO-PDFs
         mlcr detect,os -j
 
   test_mlc_access_core_actions:
@@ -271,9 +271,6 @@ jobs:
       matrix:
         python-version: ["3.12", "3.8"]
         os: ["ubuntu-latest", "windows-latest", "macos-latest"]
-        exclude:
-          - os: windows-latest
-          - os: macos-latest
 
     steps:
     - uses: actions/checkout@v4
@@ -352,8 +349,34 @@ jobs:
         mlc rm cache -h
         mlc search cache -h
         mlc show cache -h
-    - name: MLC doc script
+  
+  test_mlc_script_actions:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.13", "3.8"]
+        os: ["ubuntu-latest", "windows-latest", "macos-latest"]
+        action: ["mlcr", "mlce", "mlct"]
+
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v3
+      with:
+        python-version: ${{ matrix.python-version }}
+
+    - name: Configure git longpaths (Windows)
+      if: matrix.os == 'windows-latest'
+      run: |
+        git config --system core.longpaths true
+
+    - name: Install mlcflow from the pull request's source repository and branch
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install --ignore-installed --verbose pip setuptools
+        python -m pip install .
+    - name: MLC ${{matrix.action}} script
       run: |
-        mlc pull repo mlcommons@mlperf-automations
-        mlc doc script --tags=detect,cpu
-        mlc doc script --all
+        mlcp mlcommons@mlperf-automations
+        ${{matrix.action}} detect,cpu

.github/workflows/test-mlc-docker-core.yml

@@ -56,7 +56,7 @@ jobs:
 
     - name: Test --docker_cache 
       run: |
-        mlc docker run --tags=detect,os --docker_dt --docker_cache=no
+        mlcd detect,os --docker_dt --docker_cache=no
 
     - name: Test --docker_rebuild
       run: |

mlc/main.py

@@ -75,17 +75,26 @@ def search(self, i):
         #indices
 
 
-def mlcr():
-    first_arg_value = "run"
-    second_arg_value = "script"
+def mlc_expand_short(action, target = "script"):
 
     # Insert the positional argument into sys.argv for the main function
-    sys.argv.insert(1, first_arg_value)
-    sys.argv.insert(2, second_arg_value)
+    sys.argv.insert(1, action)
+    sys.argv.insert(2, target)
 
     # Call the main function
     main()
 
+def mlcr():
+    mlc_expand_short("run")
+def mlcd():
+    mlc_expand_short("docker")
+def mlce():
+    mlc_expand_short("experiment")
+def mlct():
+    mlc_expand_short("test")
+
+def mlcp():
+    mlc_expand_short("pull", "repo")
 
 
 def process_console_output(res, target, action, run_args):

pyproject.toml

@@ -47,3 +47,7 @@ py-modules = []
 [project.scripts]
 mlc = "mlc.main:main"
 mlcr = "mlc.main:mlcr"
+mlcd = "mlc.main:mlcd"
+mlce = "mlc.main:mlce"
+mlct = "mlc.main:mlct"
+mlcp = "mlc.main:mlcp"