From 2c9b26164c172b6d74d2a34fbd4a7c45681c07a1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ezgi=20=C3=87i=C3=A7ek?= <ezgi@meta.com>
Date: Tue, 13 Feb 2024 06:02:54 -0800
Subject: [PATCH] Use more precise function name for retained heap enabled
 check

Reviewed By: stepancheg

Differential Revision: D53672174

fbshipit-source-id: 557a70ddce930daa0f2711f28629424397d3348e
---
 starlark/src/environment/modules.rs    | 2 +-
 starlark/src/eval/runtime/evaluator.rs | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/starlark/src/environment/modules.rs b/starlark/src/environment/modules.rs
index 8d35ab867..6d7d53240 100644
--- a/starlark/src/environment/modules.rs
+++ b/starlark/src/environment/modules.rs
@@ -335,7 +335,7 @@ impl Module {
         }
     }
 
-    pub(crate) fn enable_heap_profile(&self, mode: RetainedHeapProfileMode) {
+    pub(crate) fn enable_retained_heap_profile(&self, mode: RetainedHeapProfileMode) {
         self.heap_profile_on_freeze.set(Some(mode));
     }
 
diff --git a/starlark/src/eval/runtime/evaluator.rs b/starlark/src/eval/runtime/evaluator.rs
index 70a91f2e9..6fb4f2a2e 100644
--- a/starlark/src/eval/runtime/evaluator.rs
+++ b/starlark/src/eval/runtime/evaluator.rs
@@ -289,10 +289,10 @@ impl<'v, 'a> Evaluator<'v, 'a> {
                 match mode {
                     ProfileMode::HeapFlameRetained => self
                         .module_env
-                        .enable_heap_profile(RetainedHeapProfileMode::Flame),
+                        .enable_retained_heap_profile(RetainedHeapProfileMode::Flame),
                     ProfileMode::HeapSummaryRetained => self
                         .module_env
-                        .enable_heap_profile(RetainedHeapProfileMode::Summary),
+                        .enable_retained_heap_profile(RetainedHeapProfileMode::Summary),
                     _ => {}
                 }