added support for ws_allocate

Singularity

@@ -13,6 +13,7 @@ From: nvidia/cuda:11.3.1-base-ubuntu20.04
     environment.yml
     runner.sh
     datamover.sh
+    ws_allocate
  
 %environment
   export DEBIAN_FRONTEND=noninteractive
@@ -93,6 +94,19 @@ From: nvidia/cuda:11.3.1-base-ubuntu20.04
   ln -s "$datamover" "${slurmtools}"dtls
   ln -s "$datamover" "${slurmtools}"dtcp
 
+  # set up the ws_allocate wrapper script
+  # enabling us to use the ws_* tools via ssh on the login node
+  mv /ws_allocate /usr/bin/
+
+  ws_exe=/usr/bin/ws_allocate
+
+  ln -s "$ws_exe" /usr/bin/ws_extend
+  ln -s "$ws_exe" /usr/bin/ws_release
+  ln -s "$ws_exe" /usr/bin/ws_register
+  ln -s "$ws_exe" /usr/bin/ws_restore
+  ln -s "$ws_exe" /usr/bin/ws_list
+  ln -s "$ws_exe" /usr/bin/ws_find
+
 #####################
 ## run python file ##
 #####################

ws_allocate

+#!/bin/bash
+exe=$(basename "$0")
+ssh -i /.ssh/id_rsa -o UserKnownHostsFile=/.ssh/known_hosts tauruslogin3 /usr/bin/"$exe" "$@"
\ No newline at end of file