Skip to content

Commit 9f36174

Browse files
committed
update example command line and clean up unused imports
1 parent 688c895 commit 9f36174

3 files changed

Lines changed: 5 additions & 7 deletions

File tree

predicators/approaches/vlm_online_invention_approach.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""Example command line: export OPENAI_API_KEY=<your API key>
22
3-
Example run: python scripts/run_interactive_yaml.py -c
4-
vlm_predicate_cover.yaml
3+
Example run:
4+
PYTHONPATH=. python scripts/run_interactive_yaml.py -c nsp/nsps.yaml
55
"""
66
import ast
77
import sys

predicators/nsrt_learning/strips_learning/clustering_learner.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from collections import defaultdict
77
from copy import deepcopy
88
from typing import Dict, FrozenSet, Iterator, List, Set, Tuple, cast
9-
from pympler import asizeof
109

1110
from predicators import utils
1211
from predicators.nsrt_learning.strips_learning import BaseSTRIPSLearner

predicators/utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@
5252
from scipy.stats import beta as BetaRV
5353
from tabulate import tabulate
5454
from tqdm import tqdm
55-
from pympler import asizeof
5655

5756
from predicators.args import create_arg_parser
5857
from predicators.image_patch_wrapper import ImagePatch
@@ -2786,9 +2785,9 @@ def _run_heuristic_search(
27862785
while len(queue) > 0 and time.perf_counter() - start_time < timeout and \
27872786
num_expansions < max_expansions and num_evals < max_evals:
27882787
# Checking the memory usuage of the queue
2789-
memory_usage = asizeof.asizeof(queue)
2790-
memory_usage_mb = memory_usage / (1024 * 1024)
2791-
logging.info(f"Memory usage of the candidate op: {memory_usage_mb:.2f} MB")
2788+
# memory_usage = asizeof.asizeof(queue)
2789+
# memory_usage_mb = memory_usage / (1024 * 1024)
2790+
# logging.info(f"Memory usage of the candidate op: {memory_usage_mb:.2f} MB")
27922791
_, _, node = hq.heappop(queue)
27932792
# If we already found a better path here, don't bother.
27942793
if state_to_best_path_cost[node.state] < node.cumulative_cost:

0 commit comments

Comments
 (0)