Skip to content

Commit

Permalink
Fix asdict dictionary conversion for RuntimeEnv
Browse files Browse the repository at this point in the history
  • Loading branch information
spillai committed Jan 28, 2024
1 parent 6873573 commit 32a988e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nos/managers/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import os
import re
from collections import OrderedDict
from dataclasses import asdict, dataclass, field
from dataclasses import dataclass, field
from enum import Enum
from pathlib import Path
from typing import Any, Dict, Iterable, List, Union
Expand Down Expand Up @@ -237,7 +237,7 @@ def _get_actor_options(cls, spec: ModelSpec, deployment: ModelDeploymentSpec) ->

if spec.runtime_env is not None:
logger.debug("Using custom runtime environment, this may take a while to build.")
actor_opts["runtime_env"] = RuntimeEnv(**asdict(spec.runtime_env))
actor_opts["runtime_env"] = RuntimeEnv(**spec.runtime_env.model_dump())
logger.debug(f"Actor options [id={spec.id}, opts={actor_opts}]")

return actor_opts
Expand Down

0 comments on commit 32a988e

Please sign in to comment.