feat: Add support for matrix globs.

src/rapids_dependency_file_generator/rapids_dependency_file_generator.py

@@ -1,3 +1,4 @@
+import fnmatch
 import itertools
 import os
 import textwrap
@@ -300,10 +301,11 @@ def should_use_specific_entry(matrix_combo, specific_entry_matrix):
     [matrices] list. This function validates the [matrices.matrix] value
     against the provided `matrix_combo` to check if they are compatible.
 
-    A `specific_entry_matrix` is compatible with a `matrix_combo` if and only if
-    `specific_entry_matrix[key] == matrix_combo[key]` for every key defined in
-    `specific_entry_matrix`. A `matrix_combo` may contain additional keys not
-    specified by `specific_entry_matrix`.
+    A `specific_entry_matrix` is compatible with a `matrix_combo` if and only
+    if `specific_entry_matrix[key]` matches the glob pattern
+    `matrix_combo[key]` for every key defined in `specific_entry_matrix`. A
+    `matrix_combo` may contain additional keys not specified by
+    `specific_entry_matrix`.
 
     Parameters
     ----------
@@ -321,7 +323,8 @@ def should_use_specific_entry(matrix_combo, specific_entry_matrix):
         `matrix_combo` and False otherwise.
     """
     return all(
-        matrix_combo.get(specific_key) == specific_value
+        specific_key in matrix_combo
+        and fnmatch.fnmatch(matrix_combo[specific_key], specific_value)
         for specific_key, specific_value in specific_entry_matrix.items()
     )
 

tests/examples/matrix-glob/dependencies.yaml

@@ -0,0 +1,30 @@
+files:
+  dev:
+    output: conda
+    conda_dir: output/actual
+    matrix:
+      cuda: ["10.0", "11.8", "12.0"]
+    includes:
+      - build
+channels:
+  - rapidsai
+  - conda-forge
+dependencies:
+  build:
+    common:
+      - output_types: [conda]
+        packages:
+          - clang-tools=11.1.0
+    specific:
+      - output_types: [conda]
+        matrices:
+          - matrix:
+              cuda: "11.*"
+            packages:
+              - cudatoolkit=11.*
+          - matrix:
+              cuda: "12.*"
+            packages:
+              - cuda-version=12.*
+          - matrix:
+            packages:

tests/examples/matrix-glob/output/expected/dev_cuda-100.yaml

@@ -0,0 +1,8 @@
+# This file is generated by `rapids-dependency-file-generator`.
+# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
+channels:
+- rapidsai
+- conda-forge
+dependencies:
+- clang-tools=11.1.0
+name: dev_cuda-100

tests/examples/matrix-glob/output/expected/dev_cuda-118.yaml

@@ -0,0 +1,9 @@
+# This file is generated by `rapids-dependency-file-generator`.
+# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
+channels:
+- rapidsai
+- conda-forge
+dependencies:
+- clang-tools=11.1.0
+- cudatoolkit=11.*
+name: dev_cuda-118

tests/examples/matrix-glob/output/expected/dev_cuda-120.yaml

@@ -0,0 +1,9 @@
+# This file is generated by `rapids-dependency-file-generator`.
+# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
+channels:
+- rapidsai
+- conda-forge
+dependencies:
+- clang-tools=11.1.0
+- cuda-version=12.*
+name: dev_cuda-120