Add support for ractor/ractor-only in burn_in.rb

I'm not entirely sure what this script is for. But it should probably
support all the same benchmarks as the main benchmark runner

Author: eightbitraptor

burn_in.rb

@@ -41,7 +41,7 @@
     args.num_long_runs = v.to_i
   end
 
-  opts.on("--category=headline,other,micro", "when given, only benchmarks with specified categories will run") do |v|
+  opts.on("--category=headline,other,micro,ractor,ractor-only", "when given, only benchmarks with specified categories will run") do |v|
     args.categories = v.split(",")
   end
 
@@ -70,7 +70,7 @@ def run_benchmark(bench_id, no_yjit, logs_path, run_time, ruby_version)
 
   script_path = File.join(bench_dir, bench_name, 'benchmark.rb')
   if not File.exist?(script_path)
-    script_path = File.join('benchmarks', bench_name + '.rb')
+    script_path = File.join(bench_dir, bench_name + '.rb')
   end
 
   # Assemble random environment variable options to test
@@ -198,11 +198,38 @@ def test_loop(bench_names, no_yjit, logs_path, run_time, ruby_version)
 
 # Extract the names of benchmarks in the categories we want
 metadata = YAML.load_file('benchmarks.yml')
-metadata = metadata.filter do |bench_name, entry|
-  category = entry.fetch('category', 'other')
-  args.categories.include? category
+bench_names = []
+
+if args.categories.include?('ractor-only')
+  # Only include benchmarks with ractor/ prefix (from benchmarks-ractor directory)
+  bench_names = metadata.keys.select { |name| name.start_with?('ractor/') }
+elsif args.categories.include?('ractor')
+  # Include both ractor/ prefixed benchmarks and those with ractor: true
+  metadata.each do |name, entry|
+    if name.start_with?('ractor/') || entry['ractor']
+      bench_names << name
+    end
+  end
+
+  # Also include regular category benchmarks if other categories are specified
+  if args.categories.any? { |cat| ['headline', 'other', 'micro'].include?(cat) }
+    metadata.each do |name, entry|
+      category = entry.fetch('category', 'other')
+      if args.categories.include?(category) && !bench_names.include?(name)
+        bench_names << name
+      end
+    end
+  end
+else
+  # Regular category filtering
+  metadata.each do |name, entry|
+    category = entry.fetch('category', 'other')
+    if args.categories.include?(category)
+      bench_names << name
+    end
+  end
 end
-bench_names = metadata.map { |name, entry| name }
+
 bench_names.sort!
 
 # Fork the test processes