Setting up gha

.github/workflows/train.yaml

@@ -0,0 +1,33 @@
+name: Train models
+
+on:
+  pull_request:
+    branches:
+      - main
+
+concurrency:
+  group: pull_request-${{ github.head_ref }}
+  cancel-in-progress: true
+
+
+jobs:
+    trk-2016MagUp-Sim10b:
+        uses: LamarrSim/release-models/.github/workflows/release.yaml@main
+        permissions:
+          contents: write
+          pull-requests: write
+        with:
+            name: pp-2016-MU-Sim10b-${{ github.head_ref }}
+            snakemake_dir: notebooks
+            target: deploy validate_acceptance validate_efficiency validate_resolution validate_covariance
+            model_prefix: trk
+            additional_config: |
+                storage_folder: trk-2016MagUp-Sim10b
+                training_data:
+                    - training-data/2016MU-sim10b/LamarrTraining.root
+            additional_profile: |
+                rerun-triggers: mtime
+            generated_model_path: /tmp/compiled_model.c
+        secrets:
+            SECRETS_YAML: ${{ secrets.STORAGE_SECRETS }}
+

.gitignore

@@ -1,4 +1,5 @@
 *.swp
+**/secrets.yaml
 *.swo
 *.html
 *.log

LbTrksimTrain/__main__.py

@@ -23,6 +23,7 @@
 parser.add_argument ( 'commands', nargs = "*", default = 'help', 
     choices = [
       'help',
+      'propagation',
       'acceptance', 
       'genlevelcheck', 
       'efficiency', 
@@ -54,6 +55,16 @@
 if args.verbose: 
   logging.getLogger().setLevel(logging.DEBUG) 
 
+################################################################################
+###  P R O P A G A T I O N            
+################################################################################
+if 'propagation' in args.commands:
+  from LbTrksimTrain.steps import propagation 
+  with Report ("Propagation", args.report) as report: 
+    plt.inferno() 
+    propagation.plot ( cfg, report ) 
+
+
 ################################################################################
 ###  A C C E P T A N C E  
 ################################################################################

LbTrksimTrain/config/2016-MagDown.yaml

@@ -12,8 +12,19 @@ datasets:
       #- /pclhcb06/landerli/TrackingParametrization/data/trackRes_10000000_2016_MagDown.root
       - /pclhcb06/landerli/TrackingParametrization/data/unmerged/53/*/trackRes_10000000_2016_MagDown.root
 
+    files_train:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/53/*[1-9]/trackRes_10000000_2016_MagDown.root
+
+    files_validate:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/53/*0/trackRes_10000000_2016_MagDown.root
+
+
 
   BrunelGenerated:
     files:
       - /pclhcb06/landerli/TrackingParametrization/data/unmerged/53/*/trackRes_10000000_2016_MagDown.root
+    files_train:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/53/*[1-9]/trackRes_10000000_2016_MagDown.root
+    files_validate:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/53/*0/trackRes_10000000_2016_MagDown.root
 

LbTrksimTrain/config/2016-MagUp.yaml

@@ -11,9 +11,19 @@ datasets:
     files:
       #     - /pclhcb06/landerli/TrackingParametrization/data/trackRes_10000000_2016_MagUp.root
       - /pclhcb06/landerli/TrackingParametrization/data/unmerged/52/*/trackRes_10000000_2016_MagUp.root
+
+    files_train:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/52/*[1-9]/trackRes_10000000_2016_MagUp.root
+
+    files_validate:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/52/*0/trackRes_10000000_2016_MagUp.root
 
 
   BrunelGenerated:
     files:
       - /pclhcb06/landerli/TrackingParametrization/data/unmerged/52/*/trackRes_10000000_2016_MagUp.root
+    files_train:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/52/*[1-9]/trackRes_10000000_2016_MagUp.root
+    files_validate:
+      - /pclhcb06/landerli/TrackingParametrization/data/unmerged/52/*0/trackRes_10000000_2016_MagUp.root