(self, shape_1, shape_2)
| 152 | ) |
| 153 | |
| 154 | def get_dict_example_0(self, shape_1, shape_2): |
| 155 | return { |
| 156 | "image": np.random.rand(*shape_1).astype("float32"), |
| 157 | "source": "foo", |
| 158 | "matrix": np.random.rand(*shape_2).astype("float32"), |
| 159 | } |
| 160 | |
| 161 | def get_dict_example_1(self, shape_1, shape_2): |
| 162 | return { |