画像分類のためのフェデレーテッドラーニング

Question

前提

Federated Learning for Image Classificationから
https://www.tensorflow.org/federated/tutorials/federated_learning_for_image_classification?hl=ja
Google Colabで実行をスタートさせたのですがエラーが発生いたします。

実現したいこと

import collections

import dp_accounting
import numpy as np
import pandas as pd
import tensorflow as tf
import tensorflow_federated as tff

発生している問題・エラーメッセージ

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-2-b7774dff6eec> in <module>
      5 import pandas as pd
      6 import tensorflow as tf
----> 7 import tensorflow_federated as tff

14 frames
/usr/lib/python3.8/typing.py in _type_check(arg, msg, is_argument)
    147         return arg
    148     if not callable(arg):
--> 149         raise TypeError(f"{msg} Got {arg!r:.100}.")
    150     return arg
    151 

TypeError: Callable[[arg, ...], result]: each arg must be a type. Got Ellipsis.

このようにエラーが返っていました。
Google Colabでなぜこのようなエラーが起こるのかわかりません。
お手数ですが、教えて頂けたら幸いです。

Comments

[can110]

その前の「始める前に」は正常に終了するでしょうか。
こちらで試したところ「ERROR: pip's dependency resolver does not currently take into account all the packages that are installed.」となりましたが。

[jbpb0]

can110さん

!pip install --quiet --upgrade tensorflow_federated
↓ 変更 (--quietを付けない)
!pip install --upgrade tensorflow_federated

としてから実行したら分かりますが、

> 「ERROR: pip's dependency resolver does not currently take into account all the packages that are installed.」となりましたが。

の下に
「Successfully installed absl-py-1.0.0 attrs-21.4.0 cachetools-3.1.1 contourpy-1.0.7 dm-tree-0.1.7 dp-accounting-0.3.0 farmhashpy-0.4.0 fonttools-4.38.0 grpcio-1.46.5 matplotlib-3.6.3 pandas-1.5.3 portpicker-1.5.2 semantic-version-2.6.0 tensorflow-model-optimization-0.7.3 tensorflow-privacy-0.8.7 tensorflow-probability-0.15.0 tensorflow_federated-0.33.0」
と表示されるので、「tensorflow_federated」はインストールされるようです

もちろん、「This behaviour is the source of the following dependency conflicts.」の副作用で、変なことが起きる可能性はありますが

[can110]

なるほど。
あと、たぶん同じ現象がつい最近報告されてますね。
Callable[[arg, ...], result]: each arg must be a type. Got Ellipsis. #3519
https://github.com/tensorflow/federated/issues/3519
最近Colabが20.04に変わった影響なのかなあ？

[jbpb0]

これです
https://github.com/googlecolab/colabtools/issues/3281
「The latest version of tensorflow-federated requires Python 3.9+.」

[can110]

なるほど。tensorflow-federated側の更新が原因なのですね。

[Jun0908]

can110さん、jbpb0さん
教えて頂きありがとうございます！
本当に感謝です。

Answer 1

!pip install --quiet --upgrade tensorflow_federated

↓ 変更

!pip install --quiet --upgrade tensorflow_federated==0.20.0
!pip install --quiet --upgrade tensorflow-probability==0.16.0

で、

TypeError: Callable[[arg, ...], result]: each arg must be a type. Got Ellipsis.

は出なくなると思います

参考
import tensorflow_federated as tff error in google colab #3281

Answer 2

!pip install --quiet --upgrade tensorflow_federated

↓変更

!pip install --quiet --upgrade tensorflow_federated==0.20.0
!pip install --quiet --upgrade tensorflow-probability==0.16.0
!pip install --quiet --upgrade nest-asyncio

import nest_asyncio
nest_asyncio.apply()

にしたらエラーが出なくなりました。