Skip to content

Commit

Permalink
update download
Browse files Browse the repository at this point in the history
Signed-off-by: chengtcc <864261919@qq.com>
  • Loading branch information
chengtcc committed Dec 16, 2023
1 parent 7ea7676 commit ad97d7c
Show file tree
Hide file tree
Showing 3 changed files with 24 additions and 3 deletions.
6 changes: 3 additions & 3 deletions python/eggroll/cli/commands/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,9 +138,9 @@ def download(ctx, **kwargs):
client: EggrollClient = ctx.obj["client"]
download_dir = kwargs.get("download_dir")
client._session_id = kwargs.get("session_id")
status = client.query_status()
if status["message"]:
return prettify(status)
response = client.query_status()
if response.get("message", None):
return prettify(response)

os.makedirs(download_dir, exist_ok=True)
with tempfile.TemporaryDirectory() as temp_dir:
Expand Down
Empty file.
21 changes: 21 additions & 0 deletions python/eggroll/deepspeed/utils/params_utils.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
#
# Copyright 2019 The FATE Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

def filter_invalid_params(self, value=None, **kwargs):
_p = {}
for _k, _v in kwargs.items():
if _v is not value:
_p[_k] = _v
return _p

0 comments on commit ad97d7c

Please sign in to comment.