format

COS-IN · Aug 30, 2024 · 9ea2f69 · 9ea2f69
1 parent a58ae08
commit 9ea2f69
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/Ilúvatar/iluvatar_worker_library/src/services/containers/docker/docker.rs b/src/Ilúvatar/iluvatar_worker_library/src/services/containers/docker/docker.rs
@@ -286,7 +286,10 @@ impl ContainerIsolationService for DockerIsolation {
             Ok(p) => p,
             Err(e) => return err_val(e, device_resource),
         };
-        let gunicorn_args = format!("GUNICORN_CMD_ARGS=--workers=1 --timeout={} --bind=0.0.0.0:{}", &self.limits_config.timeout_sec, port);
+        let gunicorn_args = format!(
+            "GUNICORN_CMD_ARGS=--workers=1 --timeout={} --bind=0.0.0.0:{}",
+            &self.limits_config.timeout_sec, port
+        );
         env.push(gunicorn_args.as_str());
         let mut ports = HashMap::new();
         ports.insert(
@@ -304,10 +307,10 @@ impl ContainerIsolationService for DockerIsolation {
                 Ok(p) => {
                     debug!(tid=%tid, "Acquired docker creation semaphore");
                     Some(p)
-                },
+                }
                 Err(e) => {
                     bail_error_value!(error=%e, tid=%tid, "Error trying to acquire docker creation semaphore", device_resource);
-                },
+                }
             },
             None => None,
         };

diff --git a/src/Ilúvatar/iluvatar_worker_library/src/services/resources/gpu.rs b/src/Ilúvatar/iluvatar_worker_library/src/services/resources/gpu.rs
@@ -941,9 +941,9 @@ impl GpuResourceTracker {
             }
         }
     }
-    pub fn memory_pressure(&self, gpu: &GPU) -> (MemSizeMb,MemSizeMb) {
+    pub fn memory_pressure(&self, gpu: &GPU) -> (MemSizeMb, MemSizeMb) {
         match self.gpu_metadata.get(&gpu.gpu_hardware_id) {
-            None => (0,0),
+            None => (0, 0),
             Some(meta) => {
                 let curr = *meta.device_allocated_memory.read();
                 (curr, meta.hardware_memory_mb)