From 6e1858a55e806561bda2468fd08ae2a2b4691c86 Mon Sep 17 00:00:00 2001
From: Harsh Shandilya <me@msfjarvis.dev>
Date: Thu, 8 Jun 2023 07:03:28 +0000
Subject: [PATCH] fix(benchmark): adjust for API changes

---
 .../dev/msfjarvis/claw/benchmark/BaselineProfileGenerator.kt | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/benchmark/src/main/kotlin/dev/msfjarvis/claw/benchmark/BaselineProfileGenerator.kt b/benchmark/src/main/kotlin/dev/msfjarvis/claw/benchmark/BaselineProfileGenerator.kt
index 300a3f74..2344eceb 100644
--- a/benchmark/src/main/kotlin/dev/msfjarvis/claw/benchmark/BaselineProfileGenerator.kt
+++ b/benchmark/src/main/kotlin/dev/msfjarvis/claw/benchmark/BaselineProfileGenerator.kt
@@ -6,7 +6,6 @@
  */
 package dev.msfjarvis.claw.benchmark
 
-import androidx.benchmark.macro.ExperimentalStableBaselineProfilesApi
 import androidx.benchmark.macro.junit4.BaselineProfileRule
 import androidx.test.platform.app.InstrumentationRegistry
 import androidx.test.uiautomator.UiDevice
@@ -14,7 +13,6 @@ import org.junit.Before
 import org.junit.Rule
 import org.junit.Test
 
-@OptIn(ExperimentalStableBaselineProfilesApi::class)
 class BaselineProfileGenerator {
   @get:Rule val baselineProfileRule = BaselineProfileRule()
   private lateinit var device: UiDevice
@@ -27,9 +25,8 @@ class BaselineProfileGenerator {
 
   @Test
   fun generateBaselineProfile() =
-    baselineProfileRule.collectStableBaselineProfile(
+    baselineProfileRule.collectBaselineProfile(
       packageName = PACKAGE_NAME,
-      stableIterations = 2,
       maxIterations = 8,
     ) {
       startActivityAndWait()