Skip to content

Commit 1d585f7

Browse files
Add warning for iterative prompting
1 parent 89d978d commit 1d585f7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

micro_sam/evaluation/inference.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import os
22
import pickle
3+
import warnings
34

45
from copy import deepcopy
56
from typing import Any, Dict, List, Optional, Union
@@ -517,10 +518,9 @@ def run_inference_with_iterative_prompting(
517518
n_iterations: int = 8,
518519
batch_size: int = 32,
519520
) -> None:
520-
"""
521+
"""@private"""
522+
warnings.warn("The iterative prompting functionality is not working correctly yet.")
521523

522-
Args:
523-
"""
524524
device = torch.device("cuda")
525525
model = get_trainable_sam_model(model_type, checkpoint_path)
526526

0 commit comments

Comments
 (0)