From 7535a5af62bfe4b2bba0dd927610336b54ab2c57 Mon Sep 17 00:00:00 2001
From: Woze Parrot <wozeparrot@gmail.com>
Date: Tue, 30 Apr 2024 17:38:41 -0400
Subject: [PATCH] feat: move ALL_SYSTEMS

---
 tinymod/common/benchmarks.py  | 2 ++
 tinymod/plugins/benchmarks.py | 3 +--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/tinymod/common/benchmarks.py b/tinymod/common/benchmarks.py
index 13fb4d37..cd76c76a 100644
--- a/tinymod/common/benchmarks.py
+++ b/tinymod/common/benchmarks.py
@@ -26,6 +26,8 @@ def get_benchmarks(filename: str, system: str):
   "resnet": re.compile(r"\d+[ ]+(\d+\.\d+) ms run,"),
 }
 
+ALL_SYSTEMS = ["amd", "amd-train", "nvidia", "mac"]
+
 # regex, systems, skip_count, max_count
 TRACKED_BENCHMARKS = {
   # stable diffusion
diff --git a/tinymod/plugins/benchmarks.py b/tinymod/plugins/benchmarks.py
index 766b2d46..5aca067b 100644
--- a/tinymod/plugins/benchmarks.py
+++ b/tinymod/plugins/benchmarks.py
@@ -7,7 +7,7 @@
 
 import os, logging
 
-from common.benchmarks import REPO, BENCHMARKS_DIR, TRACKED_BENCHMARKS, CachedBenchmarks, filter_points
+from common.benchmarks import REPO, BENCHMARKS_DIR, TRACKED_BENCHMARKS, ALL_SYSTEMS, CachedBenchmarks, filter_points
 
 TinyMod: Client
 GUILD: Guild
@@ -24,7 +24,6 @@
 }
 CI_CHANNEL_ID = 1068993556905218128
 GITHUB_WEBHOOK_ID = 1068993579520884826
-ALL_SYSTEMS = ["amd", "amd-train", "mac", "nvidia"]
 RANGE = range(10, 100 + 1, 10)
 
 # ***** Downloading benchmarks *****