diff --git a/zsos/__init__.py b/zsos/__init__.py index 2696bf4..c3b505a 100644 --- a/zsos/__init__.py +++ b/zsos/__init__.py @@ -1,4 +1,3 @@ import frontier_exploration - import zsos.obs_transformers.resize from zsos.policy import base_policy, llm_policy diff --git a/zsos/policy/llm_policy.py b/zsos/policy/llm_policy.py index 4c14641..d198855 100644 --- a/zsos/policy/llm_policy.py +++ b/zsos/policy/llm_policy.py @@ -3,13 +3,13 @@ import numpy as np import torch -from frontier_exploration.policy import FrontierExplorationPolicy from habitat.tasks.nav.object_nav_task import ObjectGoalSensor from habitat_baselines.common.baseline_registry import baseline_registry from habitat_baselines.common.tensor_dict import TensorDict from habitat_baselines.rl.ppo.policy import PolicyActionData from torch import Tensor +from frontier_exploration.policy import FrontierExplorationPolicy from zsos.llm.llm import BaseLLM, ClientFastChat from zsos.mapping.object_map import ObjectMap from zsos.obs_transformers.resize import image_resize diff --git a/zsos/vlm/grounding_dino.py b/zsos/vlm/grounding_dino.py index ca7438b..5d647bf 100644 --- a/zsos/vlm/grounding_dino.py +++ b/zsos/vlm/grounding_dino.py @@ -3,8 +3,8 @@ import numpy as np import torch import torchvision.transforms.functional as F -from groundingdino.util.inference import load_model, predict +from groundingdino.util.inference import load_model, predict from zsos.vlm.detections import ObjectDetections from .server_wrapper import ServerMixin, host_model, send_request, str_to_image